SHAPのwaterfall図を複数配置する(How to draw multiple SHAP waterfalls)

SHAP の waterfall図を複数配置するにはこんな感じで書けます:

sample.py

fig = plt.figure()
ax0 = fig.add_subplot(141)
shap.plots.waterfall(shap_values[1],  show=False)
ax1 = fig.add_subplot(142)
shap.plots.waterfall(shap_values[2],  show=False)
ax2 = fig.add_subplot(143)
shap.plots.waterfall(shap_values[3],  show=False)
ax3 = fig.add_subplot(144)
shap.plots.waterfall(shap_values[4],  show=False)
plt.gcf().set_size_inches(20,6) 
plt.tight_layout() 
plt.show()