"""
Felix Muzny
DS 2000
Lecture 3
September 13, 2024

An example plot program with three points and
different shapes and colors!

shapes: https://matplotlib.org/stable/api/markers_api.html
colors: https://matplotlib.org/stable/gallery/color/named_colors.html
"""

import matplotlib.pyplot as plt

def main():
    print("example plot")

    # arguments are: x, y, shape
    # the color argument is optional!
    plt.plot(1, 2, "o", color="pink")
    plt.plot(2, 4, "v", color="teal")
    plt.plot(3, 6, "s", color="purple")

    # we love labeled graphs!
    plt.xlabel("number of classes")
    plt.ylabel("hours of class")
    plt.title("number of classes vs. hours of class")

    # must save before show, if we want to save with python code!
    plt.savefig("example_plot.png")
    # must do this after plotting
    plt.show()

main()