изменение масштаба участка matplotlib в python

Question

Я пытаюсь создать диаграмму с фоном в виде карты цветов в python. Я использую следующий код:

import numpy as np
from matplotlib import cm
import matplotlib.pyplot as plt

x = np.arange(1,50,0.01)
y = 1000*np.sin(x)
yarr = np.vstack((x,))
plt.imshow(yarr, extent=(min(x),max(x), min(y),max(y)), cmap=cm.hot)
plt.plot(x, y, color='cornflowerblue',lw=4)

Однако получается изображение, похожее на:

Есть ли способ сделать график «нормальным»? То есть, чтобы я мог понять, что происходит (делая область оси более «квадратной»).

Answer 1

Проблема в том, что plt.imshow() ожидает "квадратные" пиксели, чтобы сохранить соотношение сторон изображения (см. Документацию ). Используйте опцию aspect='auto', чтобы получить желаемый результат:

import numpy as np
from matplotlib import cm
import matplotlib.pyplot as plt

x = np.arange(1,50,0.01)
y = 1000*np.sin(x)
yarr = np.vstack((x,))
plt.imshow(yarr, extent=(min(x),max(x), min(y),max(y)), cmap=cm.hot, aspect="auto")
plt.plot(x, y, color='cornflowerblue',lw=4)