如何更好地定位颜色条？

Question

我正在尝试对齐两个图的右边缘并将颜色条挂在右侧。这就是我所拥有的：

import numpy as np
import matplotlib.pyplot as plt


def F(x, y):
    u = x
    v = y
    return u, v

fig = plt.figure(figsize=(8, 8), tight_layout=True)
ax1 = plt.subplot(2, 1, 1)
ax2 = plt.subplot(2, 1, 2)

x, y = np.meshgrid(np.linspace(-1, 1, 10), np.linspace(-1, 1, 10))
u, v = F(x, y)
vres = np.sqrt(u ** 2 + v ** 2)
vector_field = ax1.quiver(x, y, u, v, vres)

fig.colorbar(vector_field, ax=ax1)

plt.show()

这就是我得到的：

我已经阅读了 colorbar 文档，但找不到我想要的，有人可以帮忙吗？

提前致谢！

Answer 1

更新的解决方案：

您实际上可以在第二个子图中添加另一个颜色条并使其不可见，就像这样。 fig.colorbar(vector_field, ax=ax2).ax.set_visible(False)

完整代码：

import numpy as np
import matplotlib.pyplot as plt
def F(x, y):
    u = x
    v = y
    return u, v
fig = plt.figure(figsize=(8, 8), tight_layout=True)
ax1 = plt.subplot(2, 1, 1)
ax2 = plt.subplot(2, 1, 2)
x, y = np.meshgrid(np.linspace(-1, 1, 10), np.linspace(-1, 1, 10))
u, v = F(x, y)
vres = np.sqrt(u ** 2 + v ** 2)
vector_field = ax1.quiver(x, y, u, v, vres)
fig.colorbar(vector_field, ax=ax1)
fig.colorbar(vector_field, ax=ax2).ax.set_visible(False)
plt.show()

结果不同的旧解决方案：

我并没有预期它会起作用，但这是您需要的吗？

colorbar 中的 ax 实际上可以获取一个轴列表，而不仅仅是一个：fig.colorbar(vector_field, ax=[ax1, ax2])。不过，您还必须删除tight_layout...tight_layout=False

import numpy as np
import matplotlib.pyplot as plt
def F(x, y):
    u = x
    v = y
    return u, v
fig = plt.figure(figsize=(8, 8), tight_layout=False)
ax1 = plt.subplot(2, 1, 1)
ax2 = plt.subplot(2, 1, 2)
x, y = np.meshgrid(np.linspace(-1, 1, 10), np.linspace(-1, 1, 10))
u, v = F(x, y)
vres = np.sqrt(u ** 2 + v ** 2)
vector_field = ax1.quiver(x, y, u, v, vres)
fig.colorbar(vector_field, ax=[ax1, ax2])
plt.show()