如何迭代熊貓中的唯一值並計算關聯值的頻率

Question

我有一個包含兩列數據的數據框，如下圖所示，並且我添加了三個額外的列，對應於 3 種“類型”中的每一種。

我的目標如下：對於每個唯一的“ID”，我想計算/統計通過我的 df 出現的“類型”的數量，並將此值添加到相應的列中。

使用以下行，我可以生成每個“類型”值的計數，但我不確定如何遍歷每個唯一的“ID”，然后將“類型”計數添加到新列。

df.groupby('ID')['Type'].value_counts()

謝謝

Answer 1

將Series.unstack和DataFrame.join輸出重塑為ID列的原始輸出：

df1 = df.join(df.groupby('ID')['Type'].value_counts().unstack(fill_value=0), on='ID')

或者使用帶有join crosstab ：

df1 = df.join(pd.crosstab(df['ID'], df['Type']), on='ID')