如何將 dataframe 轉換為關於類別的元組列表？

Question

我有以下問題。 我想將 dataframe 轉換為基於類別的元組列表。 請看下面的簡單代碼：

data = {'product_id': ['5', '7', '8', '5', '30'], 'id_customer': ['1', '1', '1', '3', '3']}
df = pd.DataFrame.from_dict(data)

#desired output is:
result = [('5', '7', '8'), ('5', '30')]

請問我該怎么做？ 這個問題對我沒有幫助： Convert pandas dataframe into a list of unique tuple

Answer 1

將GroupBy.agg與tuple一起使用，例如：

print (df.groupby('id_customer', sort=False)['product_id'].agg(tuple).tolist())
print (df.groupby('id_customer', sort=False)['product_id'].apply(tuple).tolist())

print (list(df.groupby('id_customer', sort=False)['product_id'].agg(tuple)))
print (list(df.groupby('id_customer', sort=False)['product_id'].apply(tuple)))
[('5', '7', '8'), ('5', '30')]

Answer 2

使用groupby.agg ：

>>> [tuple(v) for _, v in df.groupby('id_customer')['product_id']]
[('5', '7', '8'), ('5', '30')]
>>>