В plt.colorbar(z1_plot,cax=ax1) используйте ax= вместо cax=, то есть plt.colorbar(z1_plot,ax=ax1)
plt.colorbar(z1_plot,cax=ax1)
ax=
cax=
plt.colorbar(z1_plot,ax=ax1)