python　グラフ描画コードに関して

pythonで以下のように6行1列でグラフを表示させたいと考えているのですが、
こちら、for文等を使ってもう少し効率よくシンプルなコードできないでしょうか？

アバウトな質問で申し訳ないですが、回答いただけますと助かります。よろしくお願いします。

python
1    fig = plt.figure()
2    axes1 = fig.add_subplot(6,1,1)
3    axes2 = fig.add_subplot(6,1,2)
4    axes3 = fig.add_subplot(6,1,3)
5    axes4 = fig.add_subplot(6,1,4)
6    axes5 = fig.add_subplot(6,1,5)
7    axes6 = fig.add_subplot(6,1,6)
8    
9    axes1.plot(df.index,df["aaa"])
10    axes2.plot(df.index,df["bbb"])
11    axes3.plot(df.index,df["ccc"])
12    axes4.plot(df.index,df["ddd"])
13    axes5.plot(df.index,df["eee"])
14    axes6.plot(df.index,df["fff"])
15    axes2.set_ylim(-0.2, 4.3)
16    axes3.set_ylim(-0.2, 1.2)
17    axes4.set_ylim(-0.2, 1.2)
18    axes5.set_ylim(-0.2, 1.2)
19    axes6.set_ylim(-0.2, 1.2)
20
21    axes1.set_ylabel("aaa")
22    axes2.set_ylabel("bbb")
23    axes3.set_ylabel("ccc")
24    axes4.set_ylabel("ddd")
25    axes5.set_ylabel("eee")
26    axes6.set_ylabel("fff")

行動規範の内容に同意します

回答1件

ベストアンサー

同じ処理は以下のように for 文を使ってまとめてください。

python
1import matplotlib.pyplot as plt
2import pandas as pd
3
4# 適当なデータを作る。
5x = np.linspace(-2, 2, 30)
6df = pd.DataFrame(
7    {"a": x, "b": x ** 2, "c": x ** 3, "d": x ** 4, "e": x ** 5, "f": x ** 6}
8)
9
10# 描画する。
11fig = plt.figure(figsize=(3, 12))
12for i, name in enumerate(df, 1):
13    ax = fig.add_subplot(6, 1, i)
14    ax.plot(df.index, df[name])
15    ax.set_ylabel(name)
16
17plt.show()