How can I create a scatter plot in Python?

How can I create a scatter plot in Python?

You can use the Matplotlib library to create a scatter plot. First, install it using pip install matplotlib. Then, you can use the following code:

import matplotlib.pyplot as plt

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

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 markers?

You can customize the colors and markers by adding parameters to the scatter function. For example:

plt.scatter(x, y, color='red', marker='x')

Can I add a trend line to the scatter plot?

Yes, you can add a trend line using NumPy to calculate the line of best fit. Here's how:

import numpy as np

m, b = np.polyfit(x, y, 1)
plt.plot(x, m*np.array(x) + b, color='blue')

How do I save the plot as an image?

You can save the plot using plt.savefig('filename.png') before plt.show(). This will save the plot as a PNG file.