Matplotlib グループごとにグラフを作成したい

Question

前提・実現したいこと

Python Matplotlibでグループごとにグラフを作成したいのですが、forループの方法が分りません。以下のコードの「Lot」のグループごとに3つのグラフを作成したいです。

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

1つ目のグラフはLot"A"のみですが、2つ目のグラフは"A"と"B"、3つ目のグラフは"A"と"B"と"C"が重なったものとなってしまいます。2つ目、3つ目のグラフはそれぞれ"B"、"C"のみにしたいです。

該当のソースコード

%matplotlib inline
import matplotlib.pyplot as plt
import pandas as pd

columns = ["Lot", "Process", "C1", "C2", "C3", "C4", "C5"]
index = ["0", "1", "2", "3", "4", "5",]
data = [["A", "X", 9, 8, 5, 6, 9],
        ["A", "Y", 9, 7, 3, 7, 8],
        ["B", "X", 8, 6, 4, 5, 9],
        ["B", "Y", 7, 5, 3, 2, 5],
        ["C", "X", 6, 8, 8, 5, 3],
        ["C", "Y", 5, 7, 6, 4, 2]]

df = pd.DataFrame(data=data, columns=columns, index=index)

# グラフ作成のためデータのみのデータフレームを作成
df1 = df.loc[:, "C1":"C5"]

fig, ax = plt.subplots()

for i in range(6):
    
    x  = df1.columns
    y  = df1.iloc[i]

    ax.plot(x, y, label = df.iloc[i,1])
    ax.legend(loc= 'upper left')
    
    fig.savefig("./fig_file/Fig_"+ df.iloc[i,0] +".png")

試したこと

gr = df.groupby("Lot") でグループして、「for i in gr:」　としてもエラーが出てうまくいきませんでした。

補足情報（FW/ツールのバージョンなど）

Answer 1

ループが6回まわっているので3枚のグラフにはなりません。
次々に重なっていくのは、ループの外でsubplotを作っているからです。

特に複雑なことをしたいのでなければ、データフレームのplotメソッドを使った方が見通しが良くなると思います。

>>> print(df)
  Lot Process  C1  C2  C3  C4  C5
0   A       X   9   8   5   6   9
1   A       Y   9   7   3   7   8
2   B       X   8   6   4   5   9
3   B       Y   7   5   3   2   5
4   C       X   6   8   8   5   3
5   C       Y   5   7   6   4   2
>>> group = ["A", "B", "C"]
>>> for item in group:
...     sp = df[df["Lot"]==item].drop("Lot", axis=1).set_index("Process").T.plot()
...     sp.set_title(item)
...     plt.savefig("./fig_file/Fig_"+ item +".png")
...
Text(0.5, 1.0, 'A')
Text(0.5, 1.0, 'B')
Text(0.5, 1.0, 'C')

で実行すると、以下のグラフが保存されます。

Comments

[hikaritaro]

ご教示頂いた方法で上手くいきました。シンプルで分かりやすいコードをありがとうございました！