Matplotlibによる積み上げ棒グラフ内の数値表示と欠損値削除

Question

前提・実現したいこと

①matplotlibを利用して次のPandasDataFlameを積み上げ棒グラフとして表示。

df = pd.DataFrame({'会社':["トヨタ","ホンダ","日産","スズキ","GM","その他"],
                   'SUV':[30,0,10,15,0,30],
                   'ワゴン': [0,30,0,40,45,0],
                   'セダン': [50,25,30,45,0,25]})

②棒グラフの中に数字を記入
③ただし、欠損値(0)は表示しない

発生している問題・エラーメッセージ

①は問題ないのですが
②棒グラフの中に数字を記入
③ただし、欠損値(0)は表示しない
がどうしてもうまくできません。

該当のソースコード

import pandas as pd
import matplotlib.pyplot as plt

df = pd.DataFrame({'会社':["トヨタ","ホンダ","日産","スズキ","GM","その他"],
                   'SUV':[30,0,10,15,0,30],
                   'ワゴン': [0,30,0,40,45,0],
                   'セダン': [50,25,30,45,0,25]})
df = df.set_index('会社')
ax = plt.subplot()
df.plot.bar(stacked=True, color=['black', 'green', 'red'], ax=ax)
ax.set_ylabel('金額')

for p in ax.patches:
    ax.annotate(str(p.get_height()), (p.get_x() * 1.005, p.get_height() * 1.005),color = "w")

試したこと

https://teratail.com/questions/234747
を参照してax.textでの記載も試したのですが、座標取得がどうしてもうまくできません。

https://stackoverflow.com/questions/25447700/annotate-bars-with-values-on-pandas-bar-plots

の回答にあるax.patchesとax.annotateの方がシンプルそうだったので、
試していたのですが手詰まりになってしまいました。

Answer 1

欠損値(0)は表示しない

rect.get_height() > 0 のときだけラベルを表示するようにすればいいと思います。

参考: matplotlib - 積み上げ棒グラフを作成する方法 - pystyle

import pandas as pd
import matplotlib.pyplot as plt

df = pd.DataFrame(
    {
        "SUV": [30, 0, 10, 15, 0, 30],
        "ワゴン": [0, 30, 0, 40, 45, 0],
        "セダン": [50, 25, 30, 45, 0, 25],
    },
    index=["トヨタ", "ホンダ", "日産", "スズキ", "GM", "その他"],
)

ax = plt.subplot()
df.plot.bar(stacked=True, color=["black", "green", "red"], ax=ax)
ax.set_ylabel("金額")

for rect in ax.patches:
    if rect.get_height() > 0:
        cx = rect.get_x() + rect.get_width() / 2
        cy = rect.get_y() + rect.get_height() / 2
        value = f"{rect.get_height():.0f}"
        ax.text(cx, cy, value, color="w", ha="center", va="center")

Comments

[ka1111]

迅速かつシンプルで大変助かりました。
延々とネットで回答を探し回っていたのですが、参考サイトにたどり着くことが全然できませんでした・・・。