Как добавить границы между изображениями в python

Question

Я склеиваю два графика, используя imshow, но я пытаюсь добавить границу между изображениями и не могу найти способ сделать это.

Код, который я сейчас использую:

x = np.arange(0,5,0.1)
y = np.arange(0,5,0.1)
X, Y = np.meshgrid(x, y)
a1 = X**2+Y**2
a2 = X**2+Y**2
a3 = X**2+Y**2
a = np.concatenate((a1,a2,a3), axis=1)
fig, ax = plt.subplots()
im = ax.imshow(a, cmap=cm.RdYlGn, aspect = 'auto')
plt.show()

, который производит следующее изображение:

Однако я хотел бы нарисовать две линии 'границы' для разделения изображений (то есть между первым и вторым и вторым и третье). Как мне этого добиться?

Answer 1

Вы можете использовать plt.vlines для рисования вертикальных линий:

# collect coordinates from array shapes:
a1h, a1w = a1.shape
a2h, a2w = a2.shape

plt.vlines((a1w,a1w+a2w), (0,0), (a1h, a2h))

plt.ylim(50,0)