import matplotlib.pyplot as plt
import numpy as np
fig, (ax1, ax2) = plt.subplots(nrows=2)
# Some demo data
x = np.linspace(0, 18)
y = np.linspace(0, 1)
X, Y = np.meshgrid(x, y)
Z = np.sin(X) + Y
# Something on axes 1
ax1.plot(x, np.sin(x))
# Plot the pcolormesh
pcm = ax2.pcolormesh(X, Y, Z)
# create color bar: the important part!
axColor = plt.axes([box.x0 + box.width * 1.05, box.y0, 0.01, box.height])
plt.colorbar(pcm, cax=axColor, orientation="vertical")
plt.show()
1 回答
matplotlib
自动调整轴的大小并在其旁边生成第二个轴以显示颜色条 . 我们可以提供如下第二轴:我从this SO post学到了这一点,虽然我认为上下文有些不同 .
缺点是保存的数字将有一些有趣的空白,所以你可能想要考虑替代方案,如将颜色栏定位在下方甚至是情节上 .