具有多索引和重复字符值的 Pandas pivot_table

Question

我正在尝试像这样旋转数据框

id1 id2 property value
1   2   p        v1  
1   2   p        v2

形成

id1 id2 p
1   2   v1 & v2

到目前为止我的代码是

import pandas as pd
df = pd.DataFrame(data=
    {'id1': [1, 1],
     'id2': [2, 2],
     'property': ['p', 'p'],
     'value': ['v1', 'v2']})
df = df.groupby(["id1", "id2", "property"], as_index=False).agg({'value': ' & '.join})  # 1
print(df)
df = df.pivot_table(index=["id1", "id2"], columns="property", values="value")           # 2
print(df)

输出是

   id1  id2  property  value
0    1    2  p         v1 & v2

pandas.core.base.DataError: No numeric types to aggregate

我无法理解错误消息。有什么可以聚合的？ df 在#1 之后是否应该是可旋转的形式？

Answer 1

由于默认聚合函数，DataFrame.pivot_table 中的aggfunc 是均值，因此会引发错误。将函数更改为 join 以获得正确的输出：

df = df.pivot_table(index=["id1", "id2"], 
                    columns="property", 
                    values="value",
                    aggfunc=' & '.join)  
print (df)
property        p
id1 id2          
1   2     v1 & v2

Multiindex 的最后一个列添加 DataFrame.rename_axis 用于删除列名称和 DataFrame.reset_index：

df = df.pivot_table(index=["id1", "id2"], 
                    columns="property", 
                    values="value",
                    aggfunc=' & '.join).rename_axis(None, axis=1).reset_index()
print (df)
   id1  id2        p
0    1    2  v1 & v2