使用set_array和pyplot.pcolormesh废墟图

Question

我创建了一个xx和yy矩阵,np.meshgrid并grid通过在xx和yy上运行创建了一个值矩阵.然后我使用绘制结果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]在第一次通话中使用,也应该得到相同的答案.