Matplotlib imshow() растянуть до "fit width"

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

import numpy as np
import matplotlib.pyplot as plt

A = np.random.rand(34*52).reshape(34,52)
means = np.average(A,axis=0)

plt.figure()

plt.subplot(2,1,1)
plt.imshow(A, interpolation='nearest' )

plt.subplot(2,1,2)
plt.plot(means)

plt.show()

Как я могу растянуть ширину изображения в соответствии с совпадением графиков. Таким образом, при просмотре измерений на графике пиксели souce будут находиться в столбце непосредственно над ним.

Ответы

Ответ 1

Оказывается, это так же просто, как дать aspect='auto' вызову imshow.

plt.imshow(A, interpolation='nearest', aspect='auto')