python - 将 set_array 与 pyplot.pcolormesh 一起使用会破坏图形

Question

我有一个用 np.meshgrid 创建的 xx 和 yy 矩阵，以及一个通过在 xx 和 yy 上操作创建的值的 grid 矩阵。然后我使用 graph = plt.pcolormesh(... 绘制结果并得到这个:

然后，当我尝试使用 graph.set_array(grid.ravel()) 更新绘图中的 grid 矩阵时，它会导致图形受阻。

有谁知道如何避免这种情况？

如果有帮助，这是我的完整代码:

from pylab import *
import numpy as np
import matplotlib.pyplot as plt
from obspy import read

dx = 5   # km
dt = 5   # sec
nx = 500
ny = 500

v = 3.5  # km/s
p = 1/v

t_min = np.int_(np.sqrt((nx*dx)**2 + (ny*dx)**2))/v
nt = np.int_(t_min/dt)

# Receiver position
rx = 40 * dx
ry = 40 * dx

# Signal with ones
signal = np.zeros(2*nt)
signal[0:len(signal):len(signal)/10] = 1

# Create grid:
x_vector = np.arange(0, nx)*dx
y_vector = np.arange(0, ny)*dx
xx, yy = np.meshgrid(x_vector, y_vector)

# Distance from source to grid point
rr = np.int_(np.sqrt((xx - rx)**2 + (yy - ry)**2))

# travel time grid
tt_int = np.int_(rr/v)
tt = rr/v

# Read window of signal
wlen = np.int_(t_min/dt)
signal_window = signal[0:wlen]
grid = signal_window[tt_int/dt]

ax = plt.subplot(111)
graph = plt.pcolormesh(xx, yy, grid, cmap=mpl.cm.Reds)
plt.colorbar()
plt.plot(rx, ry, 'rv', markersize=10)
plt.xlabel('km')
plt.ylabel('km')
# plt.savefig('anitestnormal.png', bbox_inches='tight')

signal_window = signal[wlen:wlen * 2]
grid = signal_window[tt_int/dt]
graph.set_array(grid.ravel())


# plt.ion()
plt.show()

Answer 1

这有点棘手……但我相信您关于尺寸的断言是正确的。它与 pcolormesh 如何创建 QuadMesh 对象有关。

文档指出:

A quadrilateral mesh is represented by a (2 x ((meshWidth + 1) * (meshHeight + 1))) numpy array coordinates

在此上下文中，meshWidth 是您的xx，meshHeight 是您的yy。当您使用 set_array 显式设置数组时，pcolormesh 希望将其直接解释为 (meshWidth x meshHeight) 四边形，因此每个维度需要少一个点。

当我测试它时，我得到了以下行为——如果你改变

graph.set_array(grid.ravel())

到

graph.set_array(grid[:-1,:-1].ravel())

你的情节会看起来像它应该的那样。

在代码中，它看起来像在对 pcolormesh 的初始调用中，如果给出了 xx 和 yy，它们实际上应该被定义为每个多一个点维度比值数组大，如果不匹配(相差 1)，则数组自动被截断一个值。因此，即使您在第一次调用中也使用了 grid[:-1,:-1]，您也应该得到相同的答案。