如何在另一列的條件下從一列中獲取值的總和

Question

對於下圖中顯示的示例數據：

如何獲得一列中出現的相似項目的數量，條件是 customer_id 相同？

ls=[]
for i in data['customer_id']:
    sum=0
    for j in data['category']:    
        if i == j[0]:
            sum+=j[1]
    ls.append(sum)

簡而言之：

[food and fruit, vegetable, bakery and bread, cookies snacks or candies, seafoods and meat] 
customer_id[0] = [4,9,5,1,0]

Answer 1

假設您的數據被加載到 pandas dataframe 中，您可以使用：

# Sample data
labels = ["a", "b", "c", "a, b, c", "b, c"]
df = pd.DataFrame({
    "customer_id": [0, 1]*10,
    "category": [labels[np.random.randint(0,len(labels ))] for i in range(20)]
})

# Count per group and pivot the rows to columns
df.groupby(['customer_id', 'category']).size().reset_index().pivot_table(
    0, ['customer_id'], 'category').fillna(0).rename_axis(
        None, axis=1).reset_index()

output：

    customer_id   a      a, b, c      b      b, c      c
0   0           2.0          1.0    1.0       5.0    1.0
1   1           2.0          0.0    4.0       1.0    3.0