changing the scale of a matplotlib plot in python

Question

I am trying to make a chart with background as a colormap in python. I am using the following code:

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)

However, this produces an image looking like:

Is there a way to make the graph look 'normal'? I.e. so that I can what is happening (making the axis area more 'square').

Lescurel · Accepted Answer · 2020-02-25 15:32:06Z

1

The problem is that plt.imshow() expects "square" pixels, in order to preserve the aspect ratio of the image (see the documentation). Use the option aspect='auto' to get the desired result :

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)

answered Feb 25, 2020 at 15:32

Lescurel

11.7k21 silver badges43 bronze badges

Sign up to request clarification or add additional context in comments.

Collectives™ on Stack Overflow

changing the scale of a matplotlib plot in python

1 Answer 1

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

1 Answer 1

Comments

Your Answer

Sign up or log in

Post as a guest

Related