Pandas Dataframe 如何合并列

Question

我有一个如图所示的 pandas 数据框。我怎样才能把它变成如下表。 （演示在 excel 中，但我只是想向您说明表格的外观 - 这个问题与从/到 excel 导入和导出数据框无关）

谢谢

Answer 1

这是不可能的。

基础pandas.DataFrame 对象是numpy 数组，它们不会按照您建议的方式对数据进行分组。因此，不能将任意列显示为分组数据。

选项 1

可以部分使用 MultiIndex 复制您想要的输出：

import pandas as pd

df = pd.DataFrame([['AAA', 8, 2, 'BBB'],
                   ['AAA', 9, 5, 'BBB'],
                   ['AAA', 10, 6, 'BBB']],
                  columns=['Name', 'Score1', 'Score2', 'PM'])

res = df.set_index(['Name', 'PM'])

结果：

          Score1  Score2
Name PM                 
AAA  BBB       8       2
     BBB       9       5
     BBB      10       6

选项 2

或者您可以在 3 列上添加一个虚拟列和 set_index：

df['dummy'] = 0
res = df.set_index(['Name', 'PM', 'dummy'])

结果：

                Score1  Score2
Name PM  dummy                
AAA  BBB 0           8       2
         0           9       5
         0          10       6

Answer 2

您拥有的数据框看起来不错。这真的取决于你想要做什么。

如果您尝试返回名称为 AAA 且 PM 为 BBB 的数据框，则应使用 pandas 查找

dfnew = df[(df.Name == 'AAA') & (df.PM == 'BBB')]