Graphs with matplotlib: Predict the Output | StemPy Lab