To create a scatter diagram in Python using Matplotlib:
import matplotlib.pyplot as plt
x_axis = ["value_1", "value_2", "value_3", ...]
y_axis = ["value_1", "value_2", "value_3", ...]
plt.scatter(x_axis, y_axis)
plt.title("title name")
plt.xlabel("x_axis name")
plt.ylabel("y_axis name")
plt.show()
Steps to Create a Scatter Diagram in Python using Matplotlib
Step 1: Install the Matplotlib package
If you haven’t already done so, install Matplotlib using the following command:
pip install matplotlib
Step 2: Gather the data for the scatter diagram
Next, gather the data to be used for the scatter diagram.
For example, let’s say that you have the following dataset:
| unemployment_rate | index_price |
| 6.1 | 1500 |
| 5.8 | 1520 |
| 5.7 | 1525 |
| 5.7 | 1523 |
| 5.8 | 1515 |
| 5.6 | 1540 |
| 5.5 | 1545 |
| 5.3 | 1560 |
| 5.2 | 1555 |
| 5.2 | 1565 |
The ultimate goal is to depict the relationship between the unemployment_rate and the index_price.
You can accomplish this goal using a scatter diagram.
Step 3: Create the scatter diagram in Python using Matplotlib
For this final step, you may use the template below in order to create a scatter diagram in Python:
import matplotlib.pyplot as plt
x_axis = ["value_1", "value_2", "value_3", ...]
y_axis = ["value_1", "value_2", "value_3", ...]
plt.scatter(x_axis, y_axis)
plt.title("title name")
plt.xlabel("x_axis name")
plt.ylabel("y_axis name")
plt.show()
For our example:
import matplotlib.pyplot as plt
unemployment_rate = [6.1, 5.8, 5.7, 5.7, 5.8, 5.6, 5.5, 5.3, 5.2, 5.2]
index_price = [1500, 1520, 1525, 1523, 1515, 1540, 1545, 1560, 1555, 1565]
plt.scatter(unemployment_rate, index_price, color="green")
plt.title("Unemployment Rate Vs Index Price", fontsize=14)
plt.xlabel("Unemployment Rate", fontsize=14)
plt.ylabel("Index Price", fontsize=14)
plt.grid(True)
plt.show()
Run the code in Python, and you’ll get the scatter diagram.
Optionally: Create the Scatter Diagram using Pandas DataFrame
So far, you have seen how to capture the dataset in Python using lists.
Optionally, you may capture the data using Pandas DataFrame. The result would be the same under both cases.
Here is the Python code using Pandas DataFrame:
import pandas as pd
import matplotlib.pyplot as plt
data = {
"unemployment_rate": [6.1, 5.8, 5.7, 5.7, 5.8, 5.6, 5.5, 5.3, 5.2, 5.2],
"index_price": [1500, 1520, 1525, 1523, 1515, 1540, 1545, 1560, 1555, 1565],
}
df = pd.DataFrame(data)
plt.scatter(df["unemployment_rate"], df["index_price"], color="green")
plt.title("Unemployment Rate Vs Index Price", fontsize=14)
plt.xlabel("Unemployment Rate", fontsize=14)
plt.ylabel("Index Price", fontsize=14)
plt.grid(True)
plt.show()
As before, you’ll get the same scatter diagram.