Python 棒グラフ スタックを行いたい

Question

下記質問(解決済)に派生しての質問になります。
https://teratail.com/questions/353631

棒グラフの見た目は変えずに、C列に2種類のデータがあるとして、グラフの中身をスタックして表示したいと考えています。よって、下記の通り挑戦してみましたが、うまくいきません… イメージとしては、上記質問で回答頂いたグラフイメージ内にC列の内容にあわせて、スタックされることを想定しているのですが、アドバイス頂けると大変助かります。

現状は以下

理想は以下になります

import pandas as pd
import matplotlib.pyplot as plt
import seaborn as sns
import numpy as num

# テストデータ
s = """A,B,C
T,Asia,2020
H,Asia,2021
N,Asia,2020![
T,EU,2021
H,EU,2020
T,USA,2021"""
df = pd.read_csv('テスト')

s = df.groupby('A').count()['B'].reset_index().set_index('A')
s = s.sort_values(['B']) # 降順に

df["2020"] = num.where(df['C']!= '2020', 0, 1)
df["2021"] = num.where(df['C']!= '2021', 0, 1)

df.plot( kind = 'barh', y = ["2020","2021"], stacked = True, color = ['r','g'], legend = "C")

plt.ylabel("Company")
plt.savefig('plot.png')

Answer 1

pandas.crosstab()関数を使うと、カテゴリごとの出現回数・頻度を算出するクロス集計を行うことができます。
第一引数に結果の行見出しとなる列名を、第二引数に結果の列見出しとなる列名を指定します。
margins引数をTrueとすることにより、各カテゴリごとの小計および全体の総計がデータフレームに追加されます。
参考：pandasのcrosstabでクロス集計（カテゴリ毎の出現回数・頻度を算出） | note.nkmk.me

>>> from io import StringIO
>>> import pandas as pd
>>> import matplotlib.pyplot as plt
>>> s = """Company,Area,Year
... T,Asia,2020
... H,Asia,2021
... N,Asia,2020
... T,EU,2021
... H,EU,2020
... T,USA,2021"""
>>> df = pd.read_csv(StringIO(s))
>>> df
  Company  Area  Year
0       T  Asia  2020
1       H  Asia  2021
2       N  Asia  2020
3       T    EU  2021
4       H    EU  2020
5       T   USA  2021
>>> pd.crosstab(df['Company'], df['Year'], margins=True)
Year     2020  2021  All
Company                 
H           1     1    2
N           1     0    1
T           1     2    3
All         3     3    6

グラフ表示のために小計列でソートを行った後、グラフ表示には不要な小計行・小計列を削除します。

>>> pd.crosstab(df['Company'], df['Year'], margins=True).sort_values(by='All')
Year     2020  2021  All
Company                 
N           1     0    1
H           1     1    2
T           1     2    3
All         3     3    6
>>> pd.crosstab(df['Company'], df['Year'], margins=True).sort_values(by='All').drop(index='All', columns='All')
Year     2020  2021
Company            
N           1     0
H           1     1
T           1     2

DataFrame.plot.barh(stacked=True)でグラフ表示を行います。

>>> pd.crosstab(df['Company'], df['Year'], margins=True).sort_values(by='All').drop(index='All', columns='All').plot.barh(stacked=True, color=['r','g'])
<AxesSubplot:ylabel='Company'>
>>> plt.show()

Comments

[PU-TARO]

ご丁寧に解説頂き深謝申し上げます。まさにイメージ通りになります。リンク先含めて大変参考になりました。有難うございます！