Pandas groupby 和 pivot 表格绘图

Question

I have a data set as such and I want to plot a bar graph showing popluation differences between rural and urban states of all states mentioned.

I tried using 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()

But the output I get is as -

Please help me out with this code. What I expect is something like-

With male female replaced with Rural Urban

Link to the data set- https://github.com/Irene-123/Data-sets/blob/main/state_dist_sc.xls

Answer 1

Remove unwanted total row from the data and get the rows which are states

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

Create a column for the total population

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

Plot the population in Urban and Rural based on Name column

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