How can I create a scatter plot using Matplotlib?

How can I create a scatter plot using Matplotlib?

To create a scatter plot in Matplotlib, you can use the scatter() function. Here's a simple example:

import matplotlib.pyplot as plt

# Sample data
x = [1, 2, 3, 4, 5]
y = [2, 3, 5, 7, 11]

# Create scatter plot
plt.scatter(x, y)
plt.title('Scatter Plot Example')
plt.xlabel('X-axis')
plt.ylabel('Y-axis')
plt.show()

What if I want to customize the colors and sizes of the points?

You can customize the colors and sizes by passing additional parameters to the scatter() function. For example:

plt.scatter(x, y, color='red', s=100)  # 's' is the size of the points

Can I add a legend to the scatter plot?

Yes, you can add a legend by using the label parameter in the scatter() function and then calling plt.legend(). Here's how:

plt.scatter(x, y, color='red', s=100, label='Data Points')
plt.legend()

How do I save the plot as an image file?

You can save the plot using the savefig() function. For example:

plt.savefig('scatter_plot.png')