将多级索引DataFrame的行合并为逗号分隔列表

Question

给定一个多重索引的DataFrame，我想将重复的索引对组合起来，并将它们的值列为逗号分隔的列表。例如，输入：

df = pd.DataFrame({'Last Name' : ['Deere','Deere','Foo'   ,'Foo'  ,'Man'   ],
                   'First Name': ['John' ,'Jane' ,'Kung'  ,'Kung' ,'Karate'],
                   'Value1':     [ 1     , 2     , 3      , 4     , 5      ],
                   'Value2':     ['Green','Blue' ,'Yellow','Black','Purple']})

df.set_index(['Last Name','First Name'],inplace=True)

提供:

                          Value1    Value2
Last Name   First Name      
Deere       John          1         Green
            Jane          2         Blue
Foo         Kung          3         Yellow
            Kung          4         Black
Man         Karate        5         Purple

我可以帮您将其转换为以下DataFrame：

                          Value1    Value2
Last Name   First Name      
Deere       John          1         Green
            Jane          2         Blue
Foo         Kung          3,4       Yellow,Black
Man         Karate        5         Purple

- Bryce93

Answer 1

您可以先通过astype将Value1列转换为string，然后使用groupby按Last Name和First Name级别分组，并使用aggjoin来汇总：

df['Value1'] = df['Value1'].astype(str)
result = df.groupby(level=['Last Name','First Name'], sort=False).agg( ','.join)
print result
                     Value1        Value2
Last Name First Name                     
Deere     John            1         Green
          Jane            2          Blue
Foo       Kung          3,4  Yellow,Black
Man       Karate          5        Purple

- jezrael