如何在 Pandas 中分組、計數然后繪制條形圖？

Question

我有一個如下所示的Pandas數據框。

year  month  class
----  -----  -----
2015  1      1
2015  1      1
2015  1      2
2015  1      2
...

我希望能夠在一個圖上創建此數據的 2 個條形圖系列。 如果我可以做一個groupby ， count並最終得到一個data frame那么我想我可以做一個簡單的dataframe.plot.barh 。

我嘗試過的是以下代碼。

x = df.groupby(['year', 'month', 'class'])['class'].count()

x最終是一個Series 。 那么我執行以下操作以獲得DataFrame 。

df = pd.DataFrame(x)

這讓我非常接近。 數據最終如下所示。

clazz
year month clazz        
2015 1     1            2
     2     1           15
     2     2           45

但是當我做一個條形圖df.plot.bar() ，我只看到一個系列。 所需的輸出只是一個系列，從 2015-01 到 2019-12， class 1 class每月發生多少次？ 然后是另一個系列，從 2015-01 到 2019-12， class 2 class每月發生多少次？

關於如何以這種方式操縱數據的任何想法？

Answer 1

groupby - unstack應該可以解決問題：

數據

df = pd.DataFrame([[2015, 1, 1],
                    [2015, 1, 1],
                    [2015, 1, 2],
                    [2015, 1, 2],
                    [2015, 1, 2],
                    [2015, 2, 1],
                    [2015, 2, 1],
                    [2015, 2, 1],
                    [2015, 2, 2],
                    [2015, 2, 2]], columns = ['year', 'month', 'class'])

解決方案

df_gb = df.groupby(['year', 'month', 'class']).size().unstack(level=2)

輸出

df_gb.plot(kind = 'bar')

Answer 2

我們也可以使用DataFrame.pivot_table ：

df.pivot_table(index=['year','month'],columns='class',aggfunc='size').plot(kind='bar')

---

或者

df.pivot_table(index='class',columns=['year','month'],aggfunc='size').plot(kind='bar')