def add_color_bar(ax, img):
divider = make_axes_locatable(ax)
cax = divider.append_axes('right', size='7%', pad=0.1)
bar = plt.colorbar(img, cax=cax)
return bar
微信
公众号
扫码关注公众号