We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 183bbef commit b84d993Copy full SHA for b84d993
1 file changed
examples/seismic/plotting.py
@@ -29,7 +29,9 @@ def plot_perturbation(model, model1, colorbar=True):
29
domain_size = 1.e-3 * np.array(model.domain_size)
30
extent = [model.origin[0], model.origin[0] + domain_size[0],
31
model.origin[1] + domain_size[1], model.origin[1]]
32
- dv = np.transpose(model.vp.data) - np.transpose(model1.vp.data)
+ slices = tuple(slice(model.nbl, -model.nbl) for _ in range(2))
33
+ slices1 = tuple(slice(model1.nbl, -model1.nbl) for _ in range(2))
34
+ dv = np.transpose(model.vp.data[slices]) - np.transpose(model1.vp.data[slices1])
35
36
plot = plt.imshow(dv, animated=True, cmap=cm.jet,
37
vmin=min(dv.reshape(-1)), vmax=max(dv.reshape(-1)),
0 commit comments