python - 如何聚合扩展值 - python - pandas

Question

我有这样的输入:

     column1     column2
    MGI:97874   MP:0008796
    MGI:97874   MP:0009395
    MGI:97874   MP:0009937
    MGI:97874   MP:0011098
    MGI:97874   MP:0011703
    MGI:96522   MP:0001614
    MGI:96522   MP:0000364
    MGI:96522   MP:0006093

我的问题是如何将相同值的行(在第一列中)合并到一行中，因此输出将如下所示:

MGI:97874 MP:0008796 MP:0009395 MP:0009937 MP:0011098 MP:0011703
MGI:96522 MP:0001614 MP:0000364 MP:0006093

Answer 1

您需要:

g = df.groupby('column1')['column2'].apply(list).reset_index()

输出:

    column1                                                       column2
0  MGI:96522                          [MP:0001614, MP:0000364, MP:0006093]
1  MGI:97874  [MP:0008796, MP:0009395, MP:0009937, MP:0011098, MP:0011703]

现在您已经聚合了数据框，您可以以任何您想要的格式进行打印。

可能是这样的:

for idx, x in g.iterrows():
    print(x['column1'], [y for y in x['column2']])

输出:

MGI:96522 ['MP:0001614', 'MP:0000364', 'MP:0006093']
MGI:97874 ['MP:0008796', 'MP:0009395', 'MP:0009937', 'MP:0011098', 'MP:0011703']