[英]Pandas Group by sum of all the values of the group and another column as comma separated
我想按一列(標簽)分組並總結相應的數量(數量)。 相關參考號 列應以逗號分隔
import pandas as pd
tag = ['PO_001045M100960','PO_001045M100960','PO_001045MSP2526','PO_001045M870191', 'PO_001045M870191', 'PO_001045M870191']
reference= ['PA_000003', 'PA_000005', 'PA_000001', 'PA_000002', 'PA_000004', 'PA_000009']
qty=[4,2,2,1,1,1]
df = pd.DataFrame({'tag' : tag, 'reference':reference, 'qty':qty})
tag reference qty
PO_001045M100960 PA_000003 4
PO_001045M100960 PA_000005 2
PO_001045MSP2526 PA_000001 2
PO_001045M870191 PA_000002 1
PO_001045M870191 PA_000004 1
PO_001045M870191 PA_000009 1
如果我使用df.groupby('tag')['qty']。sum()。reset_index(),則會得到以下結果。
tag qty
ASL_PO_000001045M100960 6
ASL_PO_000001045M870191 3
ASL_PO_000001045MSP2526 2
我需要在參考編號處增加一列。 被添加到各個標簽下,例如,
tag qty refrence
ASL_PO_000001045M100960 6 PA_000003, PA_000005
ASL_PO_000001045M870191 3 PA_000002, PA_000004, PA_000009
ASL_PO_000001045MSP2526 2 PA_000001
我該如何實現?
謝謝。
使用pandas.DataFrame.groupby.agg
:
df.groupby('tag').agg({'qty': 'sum', 'reference': ', '.join})
輸出:
reference qty
tag
PO_001045M100960 PA_000003, PA_000005 6
PO_001045M870191 PA_000002, PA_000004, PA_000009 3
PO_001045MSP2526 PA_000001 2
注意:如果reference
', '.join
數字,則', '.join
將不起作用。 在這種情況下,請使用lambda x: ', '.join(str(i) for i in x)
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.