matplotlibの凡例がうまく表示されない

前提・実現したいこと

異なる2軸のグラフ（軸1:acc, val_acc 軸2: loss, val_loss）を作成しており、4つの項目の凡例をグラフ上に表示したいです。

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

作成したグラフに何故か凡例の代わりに小さな四角形が表示されてしまいます。添付グラフのの右上辺りに確認できます。

該当のソースコード

python
1def plot_accandloss1():
2    plt.rc('font', family='serif')
3    fig, ax1 = plt.subplots()
4    ax2 = ax1.twinx()
5    line1 = ax1.plot(hist1_acc_t, color="red")
6    line2 = ax1.plot(hist1_acc_v, color="orange")
7    line3 = ax2.plot(hist1_loss_t, color="blue")
8    line4 = ax2.plot(hist1_loss_v, color="green")
9    ax1.set_ylim([0, 1])
10    plt.title('accuracy and loss_1 ({})'.format(project_name))
11    plt.xlabel('epoch')
12    ax1.set_ylabel('accuracy')
13    ax2.set_ylabel('loss')
14    plt.legend([line1, line2, line3, line4], ['acc', 'val_acc', 'loss', 'val_loss'], ncol=2)
15    pylab.subplots_adjust(right=0.7)
16    png1 = os.path.join(graphlog, project_name + '1' + '.png')
17    eps1 = os.path.join(graphlog, project_name + '1' + '.eps')
18    plt.savefig(png1, dpi=300)  # ファイル名を合成
19    plt.savefig(eps1)  # ファイル名を合成
20    plot_show_close()
21
22plot_accandloss1()

試したこと

def plot_accandloss1():
    plt.rc('font', family='serif')
    fig, ax1 = plt.subplots()
    ax2 = ax1.twinx()
    ax1.plot(hist1_acc_t, color="red")
    ax1.plot(hist1_acc_v, color="orange")
    ax2.plot(hist1_loss_t, color="blue")
    ax2.plot(hist1_loss_v, color="green")
    ax1.set_ylim([0, 1])
    plt.title('accuracy and loss_1 ({})'.format(project_name))
    plt.xlabel('epoch')
    ax1.set_ylabel('accuracy')
    ax2.set_ylabel('loss')
    plt.legend( ['acc', 'val_acc', 'loss', 'val_loss'], ncol=2)
    pylab.subplots_adjust(right=0.7)
    png1 = os.path.join(graphlog, project_name + '1' + '.png')
    eps1 = os.path.join(graphlog, project_name + '1' + '.eps')
    plt.savefig(png1, dpi=300)  # ファイル名を合成
    plt.savefig(eps1)  # ファイル名を合成
    plot_show_close()

このコードで実行した際は以下のようなグラフとなり、凡例は表示されるのですが項目と線の色が一致しておりません。(accは赤、val_accはオレンジで表示されてほしいため)。よろしくお願いいたします。

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

ここにより詳細な情報を記載してください。

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

回答1件

ベストアンサー

プロットを行うタイミングでラベルを指定するのが一番やりやすいかと思います。

Figure.legendより存在するプロットすべてのラベルを凡例に反映できます。

matplotlib.figure.Figure — Matplotlib 3.1.1 documentation

python
1    ax1.plot(hist1_acc_t, label="acc", color="red")
2    ax1.plot(hist1_acc_v, label="val_acc", color="orange")
3    ax2.plot(hist1_loss_t, label="loss", color="blue")
4    ax2.plot(hist1_loss_v, label="val_loss", color="green")
5
6    # 中略
7
8    fig.legend(ncol=2)
9
10