python - 在 Python 中按唯一值对列进行分组

Question

我有一个包含两列的数据集，我需要将其更改为以下格式:

10  1 
10  5
10  3
11  5
11  4
12  6
12  2

对此

10  1  5  3
11  5  4
12  6  2

我需要第一列中的每个唯一值都在其自己的行中。

我是 Python 的初学者，除了阅读我的文本文件之外，我不知道如何继续。

Answer 1

您可以使用 Pandas 数据框。

import pandas as pd

df = pd.DataFrame({'A':[10,10,10,11,11,12,12],'B':[1,5,3,5,4,6,2]})
print(df)

输出:

    A  B
0  10  1
1  10  5
2  10  3
3  11  5
4  11  4
5  12  6
6  12  2

让我们使用groupby 和join:

df.groupby('A')['B'].apply(lambda x:' '.join(x.astype(str)))

输出:

A
10    1 5 3
11      5 4
12      6 2
Name: B, dtype: object