Create a scatter plot in Python with Matplotlib

import modules

import pandas as pd
import numpy as np
%matplotlib inline
import matplotlib.pyplot as plt

create dummy dataframe

raw_data = {'name': ['Willard Morris', 'Al Jennings', 'Omar Mullins', 'Spencer McDaniel'],
'age': [20, 19, 22, 21],
'favorite_color': ['blue', 'blue', 'yellow', "green"],
'grade': [88, 92, 95, 70]}
df = pd.DataFrame(raw_data, index = ['Willard Morris', 'Al Jennings', 'Omar Mullins', 'Spencer McDaniel'])
age favorite_color grade name
Willard Morris 20 blue 88 Willard Morris
Al Jennings 19 blue 92 Al Jennings
Omar Mullins 22 yellow 95 Omar Mullins
Spencer McDaniel 21 green 70 Spencer McDaniel

generate a scatter plot using matplotlib

#set up the plot
plt.scatter(df['grade'], df['age'])
plt.title('student age vs. grade')
<matplotlib.text.Text at 0x10f17ffd0>

Ace your next data science interview

Get better at data science interviews by solving a few questions per week

Find a bug? Submit a suggested change on Github, or message me on Twitter.