label class in plotting graph python code example

Example: how to plot two different class in different colour in python

groups = data.groupby("Category")
for name, group in groups:
    plt.plot(group["X Value"], group["Y Value"], marker="o", linestyle="", label=name)
plt.legend()