python - 如何在一列中创建术语计数并在 pandas 数据框中将计数作为附加值异常终止

Question

我有一个看起来像这样的 Pandas 数据框:

ID   Key
101  A
102  A
205  A
101  B
105  A
605  A
200  A
102  B

我想创建一个新表来计算“A”和“B”在“Key”列中出现的次数，并将它们作为新的两个标题。该表将如下所示:

ID  A  B 
101 1  1
102 1  1
205 1  0
105 1  0
605 1  0
200 1  0

我通过“ID”和“Key”进行了treid grouing，并获取了如下尺寸:

df.groupby(['ID', 'Key']).size().transform('A', 'B')

但它说该系列没有“transform”属性，实际上，我什至不确定是否可以将两个参数传递给“transform”

Answer 1

你很接近，需要unstack :

df = df.groupby(['ID', 'Key']).size().unstack(fill_value=0)
print (df)
Key  A  B
ID       
101  1  1
102  1  1
105  1  0
200  1  0
205  1  0
605  1  0

或者crosstab :

df = pd.crosstab(df['ID'], df['Key'])
print (df)
Key  A  B
ID       
101  1  1
102  1  1
105  1  0
200  1  0
205  1  0
605  1  0