Lollipop charts in Matplotlib - Step-by-Step Execution

Concept Flow - Lollipop charts

Start with data points

↓

Plot baseline axis

↓

Draw vertical lines from baseline to each data point

↓

Draw dots at data points

↓

Add labels and style

↓

Display lollipop chart

The flow starts with data, draws lines from a baseline to each point, then adds dots on top to create a lollipop chart.

Execution Sample

Matplotlib

import matplotlib.pyplot as plt

x = [1, 2, 3, 4]
y = [10, 15, 7, 12]

plt.stem(x, y, basefmt=" ")
plt.show()

This code creates a simple lollipop chart with vertical lines and dots for each data point.

Execution Table

Step	Action	Data/Variables	Plot Elements Created	Result
1	Define x and y data	x=[1,2,3,4], y=[10,15,7,12]	None	Data ready for plotting
2	Call plt.stem(x, y, basefmt=" ")	x, y	Vertical lines from y=0 to y[i] at x[i], dots at (x[i], y[i])	Lollipop chart elements created
3	Render plot with plt.show()	Plot elements	Figure window opens	Lollipop chart displayed
4	User views chart	Visual output	Lines and dots visible	Chart shows data points clearly

💡 Plot displayed and program waits for user to close window

Variable Tracker

Variable	Start	After Step 1	After Step 2	Final
x	undefined	[1, 2, 3, 4]	[1, 2, 3, 4]	[1, 2, 3, 4]
y	undefined	[10, 15, 7, 12]	[10, 15, 7, 12]	[10, 15, 7, 12]

Key Moments - 2 Insights

Why do we use plt.stem instead of plt.plot for lollipop charts?

plt.stem automatically draws vertical lines from the baseline to each data point and places dots, matching the lollipop chart style, as shown in execution_table step 2.

What does basefmt=" " do in plt.stem?

It removes the baseline line at y=0 so only the vertical lines and dots appear, making the chart cleaner, as seen in the plot elements created in step 2.

Visual Quiz - 3 Questions

Look at the execution_table step 2, what plot elements are created by plt.stem?

A Only dots at data points

B Vertical lines from baseline to data points and dots at data points

C Only vertical lines without dots

D A line connecting all data points

💡 Hint

See 'Plot Elements Created' column in execution_table step 2

According to variable_tracker, what is the value of y after step 1?

A [1, 2, 3, 4]

B undefined

C [10, 15, 7, 12]

D None

💡 Hint

Check the 'After Step 1' column for variable y in variable_tracker

If we remove basefmt=" " from plt.stem, what changes in the plot?

A Baseline line at y=0 appears

B Vertical lines disappear

C Dots disappear

D Plot does not display

💡 Hint

Refer to key_moments explanation about basefmt in plt.stem

Concept Snapshot

Lollipop charts use vertical lines from a baseline to data points with dots on top.
Use matplotlib's plt.stem(x, y, basefmt=" ") to create them.
basefmt=" " hides the baseline line for clarity.
They highlight individual values clearly like lollipops on sticks.

Full Transcript

Lollipop charts show data points with vertical lines from a baseline and dots on top. We start by defining x and y data arrays. Then we use matplotlib's plt.stem function to draw vertical lines from zero to each y value at x positions, and place dots at those points. The basefmt=" " argument removes the baseline line for a cleaner look. Finally, plt.show() displays the chart. This method visually emphasizes each data point individually, making it easy to compare values. The execution table traces these steps, showing data preparation, plot element creation, and display. Variable tracker confirms data values remain unchanged. Key moments clarify why plt.stem is used and the role of basefmt. The visual quiz tests understanding of plot elements, variable states, and formatting effects.

Practice

1. What is the main purpose of a lollipop chart in data visualization?

easy

A. To compare values using lines and dots for clear differences

B. To show the distribution of data using bars

C. To display data points on a map

D. To create 3D surface plots

Solution

1. Step 1: Understand lollipop chart basics

   A lollipop chart uses lines and dots to compare values clearly, making differences easy to see.

2. Step 2: Compare with other chart types

   Bar charts show distribution, maps show locations, and 3D plots show surfaces, which are different from lollipop charts.

3. Final Answer:

   To compare values using lines and dots for clear differences -> Option A

4. Quick Check:

   Lollipop chart = lines + dots comparison [OK]

Hint: Remember: lollipop = line + dot comparison [OK]

Common Mistakes:

• Confusing lollipop charts with bar charts
• Thinking lollipop charts show geographic data
• Assuming lollipop charts are 3D plots

2. Which matplotlib function is used to create a basic lollipop chart?

easy

A. plt.bar()

B. plt.scatter()

C. plt.plot()

D. plt.stem()

Solution

1. Step 1: Identify the function for lollipop charts

   Matplotlib's plt.stem() creates lollipop charts by drawing lines with markers at the ends.

2. Step 2: Differentiate from other plotting functions

   plt.bar() makes bar charts, plt.scatter() makes scatter plots, and plt.plot() makes line plots, not lollipop charts.

3. Final Answer:

   plt.stem() -> Option D

4. Quick Check:

   Use plt.stem() for lollipop charts [OK]

Hint: Use plt.stem() to combine lines and dots [OK]

Common Mistakes:

• Using plt.bar() instead of plt.stem()
• Confusing scatter plots with lollipop charts
• Trying plt.plot() which lacks markers by default

3. What will the following code output?

import matplotlib.pyplot as plt
x = [1, 2, 3]
y = [4, 2, 5]
plt.stem(x, y)
plt.show()

medium

A. A bar chart with bars at x=1,2,3 and heights 4,2,5

B. A lollipop chart with stems at x=1,2,3 and heights 4,2,5

C. A scatter plot with points at (1,4), (2,2), (3,5)

D. An error because plt.stem() needs more arguments

Solution

1. Step 1: Understand plt.stem() behavior

   The function plt.stem(x, y) draws vertical lines (stems) from zero to each y value at positions x, with dots at the top.

2. Step 2: Analyze the given data points

   Here, x = [1,2,3] and y = [4,2,5], so stems appear at x=1 with height 4, x=2 with height 2, and x=3 with height 5.

3. Final Answer:

   A lollipop chart with stems at x=1,2,3 and heights 4,2,5 -> Option B

4. Quick Check:

   plt.stem(x,y) = lollipop chart [OK]

Hint: plt.stem(x,y) draws stems at x with heights y [OK]

Common Mistakes:

• Thinking plt.stem() creates bar charts
• Confusing scatter plot with lollipop chart
• Assuming plt.stem() requires extra arguments

4. Identify the error in this code for creating a lollipop chart:

import matplotlib.pyplot as plt
x = [1, 2, 3]
y = [4, 5]
plt.stem(x, y)
plt.show()

medium

A. plt.stem() is not imported correctly

B. plt.show() is missing parentheses

C. x and y have different lengths causing a ValueError

D. No error, code runs fine

Solution

1. Step 1: Check lengths of x and y

   x has 3 elements, y has 2 elements. They must be the same length for plotting.

2. Step 2: Understand matplotlib error

   Matplotlib raises a ValueError when x and y lengths differ because it cannot pair points correctly.

3. Final Answer:

   x and y have different lengths causing a ValueError -> Option C

4. Quick Check:

   Equal length x,y needed [OK]

Hint: Check x and y lengths match before plotting [OK]

Common Mistakes:

• Ignoring length mismatch between x and y
• Thinking plt.show() needs no parentheses
• Assuming plt.stem() import causes error

5. Which code snippet correctly changes the marker color to red?

import matplotlib.pyplot as plt
x = [1, 2, 3, 4]
y = [5, 3, 9, 6]
markerline, stemlines, baseline = plt.stem(x, y)
# Change marker color to red
# Which line below is correct?

hard

A. markerline.set_color('red')

B. markerline.get_markerfacecolor()[2].set_color('red')

C. markerline.get_markerfacecolor()[2] = 'red'

D. markerline.get_markerfacecolor()[2].set_facecolor('red')

Solution

1. Step 1: Understand plt.stem() return values

   It returns markerline, stemlines, baseline. markerline is a Line2D object representing all markers.

2. Step 2: Change marker color correctly

   Use markerline.set_color('red') to change the marker colors to red.

3. Final Answer:

   markerline.set_color('red') -> Option A

4. Quick Check:

   Use set_color() on markerline to change marker colors [OK]

Hint: Use set_color() on markerline to change marker colors [OK]

Common Mistakes:

• Trying to assign color directly to a list element
• Using get_markerfacecolor() incorrectly
• Confusing set_color() with set_facecolor()