将Pandas pivot_table与margins=True一起使用时出错

Question

我的代码(来自Python数据科学手册 (O‘’Reilly)一书)：

完全披露:在撰写本书的时候，这本书还处于早期发行阶段，这意味着它仍然是未经编辑的原始版本。

import numpy as np
import pandas as pd
import seaborn as sns
titanic = sns.load_dataset('titanic')

titanic.pivot_table('survived', index='sex', columns='class')

结果是：

​

​

但是，如果我现在尝试使用margins关键字添加总计，则会发生以下错误：

titanic.pivot_table('survived', index='sex', columns='class', margins=True)

TypeError: cannot insert an item into a CategoricalIndex that is not already an existing category

知道是什么导致的吗？

版本信息：

• Python 3.4.2
• 熊猫0.16.2
• 矮胖1.9.2

Answer 1

这似乎是由于大熊猫0.15和0.16之间的变化。在以前的版本中，泰坦尼克数据集具有这样的dtype：

In [1]: import pandas, seaborn

In [2]: pandas.__version__
'0.15.2'

In [3]: titanic = seaborn.load_dataset('titanic')

In [4]: titanic.dtypes
Out[4]: 
survived         int64
pclass           int64
sex             object
age            float64
sibsp            int64
parch            int64
fare           float64
embarked        object
class           object
who             object
adult_male        bool
deck            object
embark_town     object
alive           object
alone             bool
dtype: object

新的熊猫：

In [1]: import pandas, seaborn

In [2]: pandas.__version__
'0.16.2'

In [3]: titanic = seaborn.load_dataset('titanic')

In [4]: titanic.dtypes
Out[4]: 
survived          int64
pclass            int64
sex              object
age             float64
sibsp             int64
parch             int64
fare            float64
embarked         object
class          category
who              object
adult_male         bool
deck           category
embark_town      object
alive            object
alone              bool
dtype: object

有几个列会自动转换为分类，这会导致这个错误。这本书目前还没有出版，也没有编辑；我将确保用最新版本进行测试，并在出版前修复这些类型的错误。

现在，这里有一个解决办法：

In [5]: titanic['class'] = titanic['class'].astype(object)

In [6]: titanic.pivot_table('survived', index='sex', columns='class', margins=True)
Out[6]: 
class      First    Second     Third       All
sex                                           
female  0.968085  0.921053  0.500000  0.742038
male    0.368852  0.157407  0.135447  0.188908
All     0.629630  0.472826  0.242363  0.383838

编辑:我把这个作为一个问题提交给熊猫项目：https://github.com/pydata/pandas/issues/10989