【Python】グラフの作成方法

Question

csvを取り込み、以下のようなピボットを作成しました。
ここから、図のようなグラフの書き方をご教示いただきたく。
※系統はグラフに含めず、車庫ごとに分けて作成。

# 読み込み
import pandas as pd
import numpy as np
df = pd.read_csv("train.csv", encoding="shift-jis", header=1)

# ピボット
df_pivot = pd.pivot_table(df, index=['車庫', '系統', '駅名'], values=['乗車人員'])

# グラフの作成

Answer 1

元のデータが不明なのですが、こんな感じで、df_pivot.groupby(level=0) でループ処理をするとよいのではないでしょうか

import io
import pandas as pd
import numpy as np
import matplotlib.pyplot as plt


data = """
車庫,系統,駅名,乗車人員
東京,AAA,東京,300
東京,AAA,新横浜,100
東京,AAA,熱海,50
東京,AAA,静岡,80
東京,AAA,浜松,20
大阪,BBB,大阪,200
大阪,BBB,名古屋,100
大阪,BBB,浜松,80
大阪,BBB,静岡,100
大阪,BBB,東京,0
名古屋,CCC,名古屋,200
名古屋,CCC,三河安域,70
名古屋,CCC,豊橋,60
名古屋,CCC,静岡,50
名古屋,CCC,熱海,90
名古屋,CCC,東京,0
"""

# 読み込み
df = pd.read_csv(io.StringIO(data))

# ピボット
df_pivot = pd.pivot_table(df, index=['車庫', '系統', '駅名'], values=['乗車人員'])

fig, axs = plt.subplots(2,2)
for (idx, d), ax in zip(df_pivot.groupby(level=0), axs.ravel()):
    d.index = d.index.droplevel(level=[0, 1])
    d.plot.bar(ax=ax)
    ax.set_title(idx)
plt.tight_layout()
plt.savefig('out.png')
plt.show()

Comments

[jiaetakoga]

ご回答ありがとうございます。
大変参考になりました。

復習の際、追加で疑問が出てきましたので、もしよろしければ追加でご教示いただけたら嬉しいです。

疑問：
例えば、上記のグラフで、車庫を東京だけに絞って1つのグラフだけ表示するにはどのようにしたらよろしいでしょうか。

[magichan]

tokyo_df = df_pivot[df_pivot.index.get_level_values(0) == '東京']
tokyo_df.index = tokyo_df.index.droplevel(level=[0, 1])
tokyo_df.plot.bar()
plt.show()

とか

grp_df = df_pivot.groupby(level=0)
tokyo_df = grp_df.get_group('東京')
tokyo_df.index = tokyo_df.index.droplevel(level=[0, 1])
tokyo_df.plot.bar()
plt.show()

とかでないでしょうか

[jiaetakoga]

本当にありがとうございます！