はじめに

内部でmatplotlibを使っているグラフプロット系のメソッドが出力したグラフを保存してみると、軸ラベルが見切れてしまう場合がある。
メソッド内でレイアウト設定がされているため、調節するのも一苦労だが、そこをよしなに調整する方法を書き記しておく。

やり方

pylab(matplotlib.pylabでも可)のtight_layoutメソッドを呼び出せばよい。

グラフ内の軸ラベルやメモリラベルがうまく出力できるようにグラフレイアウトが変更される。

以下は、shapモジュールのsummary_plotによる例。

shapのsummary_plotでの例

shap.summary_plot(shap_values, X, show=False)
pylab.tight_layout()
plt.savefig('./shap.png')