旋转 pandas DataFrame - AssertionError：索引长度与值不匹配

Question

我有一个 pandas.DataFrame，它不会以我期望的方式旋转。虽然pivot_table 正确安排了所有内容，但它使用聚合函数来实现这一点的事实令人反感。此外，pivot_table 似乎返回了一个不必要的复杂对象，而不是一个平面数据框。

考虑以下示例

import pandas as pd
df = pd.DataFrame({'firstname':['Jon']*3+['Amy']*2,
                   'lastname':['Cho']*3+['Frond']*2,
                   'vehicle':['bike', 'car', 'plane','bike','plane'],
                   'weight':[81.003]*3+[65.6886]*2,
                   'speed':[29.022, 95.1144, 302.952, 27.101, 344.2],})
df.set_index(['firstname','lastname','weight'])

print('------  Unnecessary pivot_table does averaging  ------')
print(pd.pivot_table(df, values='speed',
                         rows='firstname','lastname','weight'],
                         cols='vehicle'))

print('------ pivot method dies  ------')
print(df.pivot( index=['firstname','lastname','weight'],
                columns='vehicle',
                values='speed'))

pivot_table 结果是

vehicle                       bike      car    plane
firstname lastname weight                           
Amy       Frond    65.6886  27.101      NaN  344.200
Jon       Cho      81.0030  29.022  95.1144  302.952

有没有办法让pivot 提供与pivot_table 命令基本相同的输出（但希望更平坦、更整洁）？如果做不到这一点，我该如何展平pivot_table 的输出？我想要的输出更像是这样的：

firstname lastname weight     bike      car    plane                           
Amy       Frond    65.6886  27.101      NaN  344.200
Jon       Cho      81.0030  29.022  95.1144  302.952

Answer 1

如果你不想聚合pivot_table，你确实需要pivot函数。但是，pivot 不适用于提供多个索引列（实际上我不知道为什么）。但是，有一个类似于 pivot 的函数 unstack，它的工作原理相同，但基于（多）索引而不是列。

所以要使用这个，你可以先在结果中将你想要的列设置为索引/列标签作为索引：

df2 = df.set_index(['firstname','lastname','weight', 'vehicle'])

然后在最后一层取消堆叠（默认），以此类推“车辆”（成为列标签）：

In [3]: df2.unstack()
Out[3]:
                             speed
vehicle                       bike      car    plane
firstname lastname weight
Amy       Frond    65.6886  27.101      NaN  344.200
Jon       Cho      81.0030  29.022  95.1144  302.952

如果您不想要多索引，您可以使用reset_index 将结果“展平”。
唯一可能遇到的问题是列也有两个级别，因此您可以先删除第一级，然后重置索引以成为一个真正扁平的数据框：

In [17]: df3 = df2.unstack()

In [18]: df3.columns = df3.columns.droplevel(0)

In [19]: df3.reset_index()
Out[19]:
vehicle firstname lastname   weight    bike      car    plane
0             Amy    Frond  65.6886  27.101      NaN  344.200
1             Jon      Cho  81.0030  29.022  95.1144  302.952