Scatter plots in Pandas - Cheat Sheet & Quick Revision

Recall & Review

beginner

What is a scatter plot used for in data science?

A scatter plot shows the relationship between two numerical variables by displaying points on a graph. Each point represents one observation with its x and y values.

beginner

How do you create a scatter plot using pandas?

Use the DataFrame.plot.scatter(x='col1', y='col2') method, where col1 and col2 are the column names for the x and y axes.

beginner

What does each point in a scatter plot represent?

Each point represents one data record with its position showing the values of two variables: one on the x-axis and one on the y-axis.

intermediate

How can color be used in scatter plots?

Color can show a third variable by assigning different colors to points based on categories or values, helping to see patterns or groups.

beginner

What is the difference between a scatter plot and a line plot?

A scatter plot shows individual points without connecting lines, focusing on the relationship between two variables. A line plot connects points to show trends over an ordered sequence.

Which pandas method creates a scatter plot?

A DataFrame.plot.scatter()

B DataFrame.plot.line()

C DataFrame.plot.bar()

D DataFrame.plot.hist()

The plot.scatter() method is used to create scatter plots in pandas.

In a scatter plot, what do the x and y axes represent?

A Time only

B Categories

C Two numerical variables

D Frequency counts

Scatter plots show the relationship between two numerical variables on the x and y axes.

How can you add color to points in a pandas scatter plot?

A Use the 'style' parameter

B Use the 'color' parameter with a single color only

C Scatter plots cannot have colors

D Use the 'c' parameter with a list or column

The 'c' parameter lets you assign colors to points based on values or categories.

What does a cluster of points in a scatter plot suggest?

A No relationship

B A group of similar data points

C Outliers

D Missing data

Clusters show groups of points with similar values, indicating possible patterns.

Which of these is NOT a typical use of scatter plots?

A Displaying frequency counts

B Identifying outliers

C Showing correlation between variables

D Visualizing clusters

Frequency counts are better shown with histograms or bar plots, not scatter plots.

Explain how to create a scatter plot using pandas and what information it shows.

Think about which columns you choose and how points appear on the graph.

Describe how color can add meaning to a scatter plot and give an example.

Imagine points colored by different types or values.

Practice

1. What does a scatter plot primarily show in a pandas DataFrame?

easy

A. The relationship between two numeric columns

B. The frequency of a single categorical column

C. The sum of all numeric columns

D. The average value of a column over time

Solution

1. Step 1: Understand scatter plot purpose

   A scatter plot is used to visualize how two numeric variables relate to each other by plotting points on an x and y axis.

2. Step 2: Match with options

   Only The relationship between two numeric columns describes this relationship between two numeric columns, while others describe different chart types or summaries.

3. Final Answer:

   The relationship between two numeric columns -> Option A

4. Quick Check:

   Scatter plot = relationship between two numeric columns [OK]

Hint: Scatter plots always compare two numeric columns [OK]

Common Mistakes:

• Confusing scatter plots with bar charts
• Thinking scatter plots show single column frequency
• Assuming scatter plots summarize data instead of showing relationships

2. Which of the following is the correct syntax to create a scatter plot in pandas for columns 'height' and 'weight'?

easy

A. df.plot.scatter('weight', 'height')

B. df.scatter_plot('height', 'weight')

C. df.plot.scatter(x='height', y='weight')

D. df.plot.scatter(x=height, y=weight)

Solution

1. Step 1: Recall pandas scatter plot syntax

   The correct method is df.plot.scatter with x and y as column names in quotes.

2. Step 2: Check each option

   df.plot.scatter(x='height', y='weight') uses correct method and argument format. df.scatter_plot('height', 'weight') is invalid method. df.plot.scatter('weight', 'height') swaps x and y but is still valid syntax, but question asks for 'height' as x and 'weight' as y. df.plot.scatter(x=height, y=weight) uses variables without quotes causing error.

3. Final Answer:

   df.plot.scatter(x='height', y='weight') -> Option C

4. Quick Check:

   Correct syntax uses df.plot.scatter with x and y as strings [OK]

Hint: Use df.plot.scatter(x='col1', y='col2') with quotes [OK]

Common Mistakes:

• Using method scatter_plot instead of plot.scatter
• Passing column names without quotes
• Swapping x and y columns unintentionally

3. Given the code:

import pandas as pd
import matplotlib.pyplot as plt

data = {'age': [25, 30, 35, 40], 'income': [50000, 60000, 65000, 70000]}
df = pd.DataFrame(data)
df.plot.scatter(x='age', y='income')
plt.show()

What will the scatter plot show?

medium

A. A bar chart of age vs income

B. An error because plt.show() is missing

C. A line plot connecting age and income

D. Points showing income increasing with age

Solution

1. Step 1: Understand the code

   The DataFrame has 'age' and 'income' columns. The scatter plot plots 'age' on x-axis and 'income' on y-axis.

2. Step 2: Interpret the plot type and data

   Scatter plot will show points representing each person's age and income. Income increases as age increases, so points trend upward.

3. Final Answer:

   Points showing income increasing with age -> Option D

4. Quick Check:

   Scatter plot shows points, not bars or lines [OK]

Hint: Scatter plots show points, not bars or lines [OK]

Common Mistakes:

• Confusing scatter plot with bar or line plot
• Thinking plt.show() is optional and causes error
• Misreading which column is x or y

4. What is wrong with this code snippet to create a scatter plot?

df.plot.scatter(x=age, y='income')

medium

A. Scatter plots require plt.scatter, not df.plot.scatter

B. The variable age is not defined or quoted

C. plot.scatter does not exist in pandas

D. The y parameter should be numeric, not string

Solution

1. Step 1: Check the x parameter

   The x parameter should be the column name as a string, but age is unquoted, treated as a variable.

2. Step 2: Verify other parameters and method

   y='income' is correct as a string. plot.scatter is valid pandas method. So only x parameter is wrong.

3. Final Answer:

   The variable age is not defined or quoted -> Option B

4. Quick Check:

   Column names must be strings in plot.scatter [OK]

Hint: Always quote column names in plot.scatter [OK]

Common Mistakes:

• Passing column names without quotes
• Confusing pandas plot.scatter with matplotlib scatter
• Thinking y must be numeric type, not string

5. You have a DataFrame with columns 'temperature', 'sales', and 'day'. You want to create a scatter plot showing sales vs temperature, but color points by day of the week. Which approach is correct?

hard

A. Use df.plot.scatter(x='temperature', y='sales', c=df['day'].astype('category').cat.codes)

B. Use df.plot.scatter(x='temperature', y='sales', color='day')

C. Use df.plot.scatter(x='temperature', y='sales', c='day')

D. Use df.plot.scatter(x='temperature', y='sales', c=df['day'])

Solution

1. Step 1: Understand color parameter in scatter plot

   The c parameter expects numeric values for colors. The 'day' column is categorical (strings), so it must be converted to numeric codes.

2. Step 2: Check each option's color argument

   Use df.plot.scatter(x='temperature', y='sales', c=df['day'].astype('category').cat.codes) converts 'day' to category codes, which is correct. Use df.plot.scatter(x='temperature', y='sales', color='day') uses color='day' which is invalid. Use df.plot.scatter(x='temperature', y='sales', c='day') passes string 'day' which is invalid. Use df.plot.scatter(x='temperature', y='sales', c=df['day']) passes categorical strings directly, causing error.

3. Final Answer:

   Use df.plot.scatter(x='temperature', y='sales', c=df['day'].astype('category').cat.codes) -> Option A

4. Quick Check:

   Color values must be numeric codes for categorical data [OK]

Hint: Convert categorical to numeric codes for color in scatter [OK]

Common Mistakes:

• Passing categorical strings directly to color parameter
• Using color='day' instead of numeric codes
• Not converting categories before coloring