python - Pandas 将数据框转换为具有一些特殊要求的元组数组

Question

我正在使用 Apriori 算法分析一些数据。这需要我将数据帧转换为元组数组(groupby col“c1”)，每个元组对应于数据帧的“行”。如果元组中只有一个值，我想在它后面加一个“,”，因为我需要将它作为一个元组保存。

In [1]: data
Out[1]:
   c1   c2
0  r1   aa
1  r1   bb
2  r1   cc
3  r2   dd
4  r2   ee
5  r3   ff

我希望这样的数据:

[('aa','bb','cc'),('dd','ee'),('ff',)]

Answer 1

使用GroupBy.apply使用 tuple 并将 Series 转换为 list:

a = df.groupby('c1')['c2'].apply(tuple).tolist()
print (a)
[('aa', 'bb', 'cc'), ('dd', 'ee'), ('ff',)]