Matplotlib intro: line plots

line_plot_demo.py

from matplotlib import pyplot as plt

nums = [1, 2, 3, 4, 5, 6, 7, 8, 9, 10]
doubles = [2, 4, 6, 8, 10, 12, 14, 16, 18, 20]
squares = [1, 4, 9, 16, 25, 36, 49, 64, 81, 100]

plt.plot(nums, doubles, label="Doubles", color="red")
plt.plot(nums, squares, label="Squares", color="blue")

plt.xlabel("Number")
plt.ylabel("Result")
plt.title("Evolution of doubles and squares for the range [1,10]")
plt.legend()
plt.tight_layout()
plt.savefig("line_plot_demo.png")
plt.show()