使用set_array和pyplot.pcolormesh废墟图

Question

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

然后，当我尝试使用grid更新绘图中的graph.set_array(grid.ravel())矩阵时，会导致数字被挫败。

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

以下是我的完整代码，如果它有帮助：

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对象有关。

文件说明：

  

四边形网格由（2 x（（meshWidth + 1）*（meshHeight + 1）））numpy数组坐标

表示

在此上下文中，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]，也应该得到相同的答案。