Matplotlib imshow() estirar para "ajustar el ancho"

Question

Tengo una imagen y una medida asociada con cada columna de sus píxeles. Estoy usando pyplot para crear una figura con la imagen en la parte superior, y un gráfico de las medidas de la columna a continuación. Estoy usando algo como esto:

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() 

¿Cómo puedo estirar el ancho de la imagen para que coincida con la de las parcelas. De esta manera, al mirar las medidas en la trama, los píxeles souce estarán en una columna directamente encima de ella.

Answer 1

Resulta que es tan simple como dar aspect='auto' a la llamada imshow.

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