pivot groupby 和 sum pandas 数据框

Question

这应该很简单，但我是使用 python 工作的新手。有什么建议吗？#original dataframedf = pd.DataFrame({'year':[1,1,1,1,1],                   'month':[4,4,4,4,4],                   'mode': ['a','b','a','a','b']},                  columns=['year','month','mode'])#pivot/groupby etc# df2=df.pivot(columns=('year','month'), values=('mode')).count()#create this dataframedf2 = pd.DataFrame({'year':[1],                   'month':[4],                   'a': [3],                   'b':[2]},                  columns=['year','month','a','b'])我在 Koalas Apache Spark 环境中工作（文档），所以解决方案应该可以解决这个问题。

Answer 1

df.pivot_table(index=['year','month'], aggfunc='size', columns='mode')




Answer 2

或者你可以使用pd.get_dummies()：

pd.get_dummies(df).groupby(['year','month']).sum()

结果：

            mode_a  mode_b

year month                

1    4           3       2

注意：我不确定它是否适用于 Koalas Apache Spark 环境。