Matplotlib imshow() stretch to “fit width”

Turns out that it’s as simple as giving aspect="auto" to the imshow call.

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