Python为多个子图设置同一个colorbar

研究了半天才整明白，直接上代码：import matplotlib.pyplot as pltimport numpy as npfig, ax = plt.subplots(2, 3)ax = ax.flatten()np.random.seed(0)for i in range(6):img = np.random.random([7, 7])im = ax[i].imshow(img)fig

Hello3q3q

8912人浏览 · 2021-12-15 20:57:36

Hello3q3q · 2021-12-15 20:57:36 发布

研究了半天才整明白，直接上代码：

import matplotlib.pyplot as plt
import numpy as np

fig, ax = plt.subplots(2, 3)
ax = ax.flatten()

np.random.seed(0)
for i in range(6):
    img = np.random.random([7, 7])
    im = ax[i].imshow(img)

fig.colorbar(im, ax=[ax[i] for i in range(6)], fraction=0.02, pad=0.05)
plt.savefig('test.png', bbox_inches='tight')
plt.show()