Как нанести изображение поверх колориметрической в ​​Matplotlib?

Question

Я хочу построить bool плотность

plt.imshow(rectangle, extent=extent, cmap=ListedColormap([[1, 1, 1, 0], [0, 0, 0, 1]]))

более colormesh

plt.pcolormesh(x_ticks, y_ticks, np.transpose(potential))

Как это сделать?

В настоящее время я вижу только ПЕРВЫЙ заговор, несмотря на то, что я ожидал чего-то еще.

---

Вот пример кода:

import matplotlib
from matplotlib.colors import ListedColormap
import matplotlib.pyplot as plt
import numpy as np

x_ticks = np.linspace(-10, 10, 100, dtype=np.float32)
y_ticks = np.linspace(-10, 10, 100, dtype=np.float32)

x, y = np.meshgrid(x_ticks, y_ticks, indexing='ij')

# density plot of some circular function
r = np.sqrt(x ** 2 + y ** 2)
z = np.where(r != 0, np.sin(r) / r, 1)

plt.figure(1)
plt.pcolormesh(x_ticks, y_ticks, z)
plt.show()

# image of rectangle, white color is transparent
rectangle = np.logical_and(np.abs(x) < 2,  np.abs(y) < 2)
extent = [x_ticks[0], x_ticks[-1], y_ticks[0], y_ticks[-1]]

plt.figure(2)
plt.imshow(rectangle, extent=extent, cmap=ListedColormap([[1, 1, 1, 0], [0, 0, 0, 1]]))
plt.show()

# now I want both on the same plot superimposed
plt.figure(3)
plt.pcolormesh(x_ticks, y_ticks, z)
plt.imshow(rectangle, extent=extent, cmap=ListedColormap([[1, 1, 1, 0], [0, 0, 0, 1]]))
plt.show()

Я получаю это:

Но мне хотелось этого (нарисовал в фотошопе):

Answer 1

Ответ Андры, конечно, правильный. Вы можете немного упростить свой код и пренебречь ключевым словом extent или zorder, если вы также используете pcolormesh для построения прямоугольника:

plt.pcolormesh(x_ticks, y_ticks, z)
plt.pcolormesh(x_ticks, y_ticks, rectangle,cmap=ListedColormap([[1, 1, 1, 0], [0, 0, 0, 1]]))

Таким образом, вы сразу узнаете, какая у вас система координат (z и rectangle должны иметь одинаковую форму).

Answer 2

Вы должны установить явный zorder в одном из вызовов печати, чтобы поместить это сверху / снизу:

plt.pcolormesh(x_ticks, y_ticks, z)
plt.imshow(rectangle, extent=extent, cmap=ListedColormap([[1, 1, 1, 0], [0, 0, 0, 1]]), zorder=1)