Pandas groupby 和 pivot 表格繪圖

Question

我有一個這樣的數據集，我想 plot 條形圖顯示所有提到的州的rural and urban州之間的人口差異。

我嘗試使用 groupby -

Literacy_States=Rural_Urban[['Name','TRU','M_LIT','F_LIT']]


Literacy_States=Literacy_States.groupby(['Name','TRU'])['M_LIT','F_LIT'].count().unstack('TRU').plot.bar()

但是我得到的 output 是-

請幫我解決這個代碼。 我期望的是類似的東西 -

以male female換Rural Urban

鏈接到數據集 - https://github.com/Irene-123/Data-sets/blob/main/state_dist_sc.xls

Answer 1

從數據中刪除不需要的總行並獲取狀態行

Literacy_States = Literacy_States[(Literacy_States.TRU != 'Total') & (Literacy_States.Level == 'State')].copy()

為總人口創建一個列

Literacy_States['population'] = Literacy_States[['M_LIT', 'F_LIT']].sum(axis=1)

Plot 基於名稱列的城鄉人口

Literacy_States.groupby(['Name', 'TRU'])['population'].sum().unstack(1).plot.bar(rot=45)