आपको matplotibib.pyplot आयात करना होगा।
pyplot you have to import matplotlib.