旋转 pandas 数据框以生成（seaborn）热图答案

【问题标题】：Pivoting a pandas dataframe to generate a (seaborn) heatmap旋转 pandas 数据框以生成（seaborn）热图
【发布时间】：2015-09-27 01:15:52
【问题描述】：

我是 Python 新手，对 SO 还很陌生。

我有一个名为 df 的 pandas 数据框，如下所示：

                                 Text
Date        Location                           
2015-07-08  San Diego, CA        1
2015-07-07  Bellevue, WA         1
            Los Angeles, CA      1
            New York, NY         1
            Los Angeles, CA      1
            Unknown              1

我想使用以下方法对数据进行透视：

import pandas, numpy as np

df_pivoted = df.pivot_table(df, values=['Text'], index=['Date'],
    columns=['Location'],aggfunc=np.sum)

这个想法是生成一个热图，按“位置”和“日期”显示“文本”的计数。

我得到错误：

TypeError: pivot_table() got multiple values for keyword argument 'values'

使用简化方法时：

df = df.pivot_table('Date', 'Location', 'Text')

我得到错误：

raise DataError('No numeric types to aggregate')

我正在使用 Python 2.7 和 Pandas 0.16.2

In[2]: df.dtypes
Out[2]: 
Date        datetime64[ns]
Text                object
Location            object
dtype: object

有人有想法吗？

【问题讨论】：

Text 列的 dtype 为 object，这可能意味着该列中的项目是字符串。它们都是数字字符串吗？或者某些行是否包含空格？ No numeric types to aggregate 建议 Text 必须是数字才能使 df.pivot_table 工作....
如果Date 和Location 组成一个MultiIndex，df.unstack(level='Location') 会简单得多。

标签： python pandas seaborn

【解决方案1】：

import pandas as pd
import numpy as np

# just try to replicate your dataframe
# ==============================================
date = ['2015-07-08', '2015-07-07', '2015-07-07', '2015-07-07', '2015-07-07', '2015-07-07']
location = ['San Diego, CA', 'Bellevue, WA', 'Los Angeles, CA', 'New York, NY', 'Los Angeles, CA', 'Unknown']
text = [1] * 6
df = pd.DataFrame({'Date': date, 'Location': location, 'Text': text})

Out[141]: 
         Date         Location  Text
0  2015-07-08    San Diego, CA     1
1  2015-07-07     Bellevue, WA     1
2  2015-07-07  Los Angeles, CA     1
3  2015-07-07     New York, NY     1
4  2015-07-07  Los Angeles, CA     1
5  2015-07-07          Unknown     1

# processing
# ==============================================
pd.pivot_table(df, index='Date', columns='Location', values='Text', aggfunc=np.sum)

Out[142]: 
Location    Bellevue, WA  Los Angeles, CA  New York, NY  San Diego, CA  Unknown
Date                                                                           
2015-07-07             1                2             1            NaN        1
2015-07-08           NaN              NaN           NaN              1      NaN

【讨论】：

非常感谢！简洁大方。