如何基于另一个列值组合int列,并将组合的int放入列表中

时间:2019-09-09 06:08:50

标签: python pandas

说我有这个熊猫数据框

dfTest = pd.DataFrame(np.random.randint(0,3,size=(15, 2)), columns=list('AB'))

dfTest.head(15)


    A   B
0   2   1
1   2   1
2   2   1
3   0   2
4   2   0
5   1   1
6   0   2
7   2   0
8   1   0
9   0   2
10  2   0
11  1   2
12  2   0
13  1   1
14  2   0

如何根据A列的值合并行,并将B列的值合并到列表中?

这是我想要的结果

    A   B
0  2  [1, 1, 1, 0, 0, 0, 0, 0]
1  0   [2, 2, 2 ]
2  1   [1, 0, 2, 1 ]

因此,共享具有Column A相同值的行将合并,但是虽然Column A仅具有共享值,但是Column中的值将被放入列表中。

0 个答案:

没有答案