how to plot graph in python in a function usinf matlab code example

Example 1: how to plot a graph using matplotlib

from matplotlib import pyplot as plt
plt.plot([0, 1, 2, 3, 4, 5], [0, 1, 4, 9, 16, 25])
plt.show()

Example 2: import pyplot python

import matplotlib.pyplot as plt