Lollipop charts in Matplotlib - Time & Space Complexity

Time Complexity: Lollipop charts

O(n)

Understanding Time Complexity

When creating lollipop charts with matplotlib, it is important to understand how the time to draw the chart changes as the data size grows.

We want to know how the number of points affects the time matplotlib takes to draw the lines and markers.

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import matplotlib.pyplot as plt

n = 10  # Example value for n
x = range(n)
y = [value for value in range(n)]

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

This code draws a lollipop chart with n points, where each point has a vertical line and a marker.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Drawing n vertical lines and n markers on the plot.
• How many times: Each line and marker is drawn once for each of the n data points.

How Execution Grows With Input

As the number of points n increases, the number of lines and markers to draw also increases proportionally.

Input Size (n)	Approx. Operations
10	About 10 lines and 10 markers drawn
100	About 100 lines and 100 markers drawn
1000	About 1000 lines and 1000 markers drawn

Pattern observation: The work grows directly with the number of points; doubling n roughly doubles the drawing work.

Final Time Complexity

Time Complexity: O(n)

This means the time to draw the lollipop chart grows linearly with the number of data points.

Common Mistake

[X] Wrong: "Drawing a lollipop chart takes the same time no matter how many points there are."

[OK] Correct: Each point adds a line and a marker, so more points mean more drawing work and longer time.

Interview Connect

Understanding how plotting time grows with data size helps you write efficient visualizations and explain performance in real projects.

Self-Check

What if we added a loop to draw multiple lollipop charts one after another? How would the time complexity change?

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()