python - pandas 中的 dcast 和聚合长度

Question

我来自 R 背景，所以努力实现我可以在 R 中轻松实现的目标，我有以下 pandas 数据框。

ID   credit_active    credit_currency      credit_type
1    Active           Dollars              Home
1    Closed           Dollars              Home
1    Active           Euro                 Home
2    Active           Dollars              Home
2    Closed           Dollars              Home
2    Active           Euro                 Home

我想要以下格式的数据

ID     Active    Closed     Dollars     Euro     Home
1      2         1          2           1        3
2      2         1          2           1        3

我可以使用 R 中的简单命令来实现这一点，如下所示，

dcast(melt(setDT(bureau), id.var = 'ID'), ID ~ value)

我怎样才能做同样的事情 Pandas ？

Answer 1

使用melt与 groupby ，合计size最后由 unstack reshape :

df = df.melt('ID').groupby(['ID', 'value']).size().unstack(fill_value=0)

另一个解决方案 pivot_table :

df = df.melt('ID').pivot_table(index='ID', columns='value', aggfunc='size')

或者使用 crosstab :

df1 = df.melt('ID')
df = pd.crosstab(df1['ID'], df1['value']) 

<小时/>

print (df)
value  Active  Closed  Dollars  Euro  Home
ID                                        
1           2       1        2     1     3
2           2       1        2     1     3