matplotlibでの複数グラフ表示について

Question

前提・実現したいこと

matplotlibを利用し複数のグラフを同時に表示しようとしているのですが、
下のコードのようにax[n][n].legend()を何度も書かねばならず面倒に感じています。
ax.legend()のような1行で6個のグラフに命令を与えたいのですがどうすればよいでしょうか

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

エラーはありません、めんどくさいだけです

該当のソースコード

# グラフ化
fig,ax =plt.subplots(3,2,figsize=(16,8))

ax[0][0].plot(t,surge,alpha=0.5,label='surge')
ax[1][0].plot(t,sway,alpha=0.5,label='sway')
ax[2][0].plot(t,heave,alpha=0.5,label='heave')

ax[0][1].plot(t,roll,alpha=0.5,label='roll')
ax[1][1].plot(t,pitch,alpha=0.5,label='pitch')
ax[2][1].plot(t,yaw,alpha=0.5,label='yaw')

ax[0][0].grid()
ax[1][0].grid()
ax[2][0].grid()
ax[0][1].grid()
ax[1][1].grid()
ax[2][1].grid()

ax[0][0].legend()
ax[1][0].legend()
ax[2][0].legend()
ax[0][1].legend()
ax[1][1].legend()
ax[2][1].legend()

plt.show()

試したこと

ここに問題に対して試したことを記載してください。

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

jupyter notebook

Answer 1

forループを使うとかなり楽になるのではないでしょうか。

for i, (data, dname) in enumerate(zip([surge, sway, heave, roll, pitch, yaw], ["surge", "sway", "heave", "roll", "pitch", "yaw"]))
    a, b = i % 3, i // 3
    ax[a][b].plot(t, data, alpha=0.5, label=dname)
    ax[a][b].grid()
    ax[a][b].legend()