If you want to draw many lines then you could use a for loop
combined with plt.plot
. You can add a legend by adding a
label
to each plot and by calling plt.legend()
afterwards.
import numpy as np
import matplotlib.pylab as plt
x = np.linspace(0, 1, 100)
for i in [1, 2, 3]:
y = x ** i
plt.plot(x, y, label=f"power={i}")
plt.legend();
Again, note the semi-colon at the end here.