Diverging bar charts in Matplotlib - Time & Space Complexity

Time Complexity: Diverging bar charts

O(n)

Understanding Time Complexity

We want to understand how the time to create a diverging bar chart changes as the data size grows.

How does the number of bars affect the work matplotlib does?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import matplotlib.pyplot as plt
import numpy as np

values = np.random.randn(100)  # 100 data points
colors = ['green' if x >= 0 else 'red' for x in values]

plt.bar(range(len(values)), values, color=colors)
plt.show()

This code creates a diverging bar chart with 100 bars colored by positive or negative values.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: Drawing each bar in the chart.
• How many times: Once for each data point (here, 100 times).

How Execution Grows With Input

As the number of bars increases, the work to draw them grows roughly in direct proportion.

Input Size (n)	Approx. Operations
10	10 drawing operations
100	100 drawing operations
1000	1000 drawing operations

Pattern observation: Doubling the number of bars roughly doubles the work.

Final Time Complexity

Time Complexity: O(n)

This means the time to draw the chart grows linearly with the number of bars.

Common Mistake

[X] Wrong: "Adding more bars won't affect the drawing time much because the chart size stays the same."

[OK] Correct: Each bar requires separate drawing work, so more bars mean more work even if the chart area is fixed.

Interview Connect

Understanding how drawing operations scale helps you explain performance in data visualization tasks clearly and confidently.

Self-Check

"What if we added nested loops to draw multiple diverging bar charts in a grid? How would the time complexity change?"

Practice

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

easy

A. To visualize hierarchical data using nested bars

B. To show positive and negative values side by side for easy comparison

C. To plot time series data with a line connecting points

D. To display only positive values in a stacked format

Solution

1. Step 1: Understand the chart type

   Diverging bar charts are designed to display data that has both positive and negative values.

2. Step 2: Identify the purpose

   They place positive and negative bars side by side to highlight differences clearly.

3. Final Answer:

   To show positive and negative values side by side for easy comparison -> Option B

4. Quick Check:

   Diverging bar chart = positive & negative side by side [OK]

Hint: Look for charts showing positive and negative bars side by side [OK]

Common Mistakes:

• Confusing diverging bar charts with stacked bar charts
• Thinking they only show positive values
• Mixing up with line charts

2. Which of the following is the correct way to create a diverging bar chart using matplotlib's barh function for negative and positive values?

easy

A. plt.barh(y_pos, abs(values), color='red')

B. plt.barh(y_pos, values, color='blue')

C. plt.barh(y_pos, values, color=['red' if v < 0 else 'green' for v in values])

D. plt.barh(y_pos, values, color=['green' if v > 0 else 'blue' for v in values])

Solution

1. Step 1: Understand color assignment for diverging bars

   Colors should differ based on whether values are positive or negative to highlight divergence.

2. Step 2: Check the color list comprehension

   plt.barh(y_pos, values, color=['red' if v < 0 else 'green' for v in values]) uses a list comprehension assigning 'red' for negative and 'green' for positive values, which is correct.

3. Final Answer:

   plt.barh(y_pos, values, color=['red' if v < 0 else 'green' for v in values]) -> Option C

4. Quick Check:

   Color by sign = plt.barh(y_pos, values, color=['red' if v < 0 else 'green' for v in values]) [OK]

Hint: Use list comprehension to color bars by sign [OK]

Common Mistakes:

• Using a single color for all bars
• Taking absolute values which lose sign info
• Assigning wrong colors to positive/negative values

3. Given the code below, what will be the color of the bar corresponding to the value -3?

import matplotlib.pyplot as plt
values = [4, -3, 2, -1]
y_pos = range(len(values))
colors = ['green' if v > 0 else 'red' for v in values]
plt.barh(y_pos, values, color=colors)
plt.show()

medium

A. No color (default)

B. Green

C. Blue

D. Red

Solution

1. Step 1: Analyze the color assignment logic

   The list comprehension assigns 'green' if value > 0, else 'red'.

2. Step 2: Check the value -3

   -3 is less than 0, so it gets 'red'.

3. Final Answer:

   Red -> Option D

4. Quick Check:

   Negative value color = red [OK]

Hint: Negative values get 'red' color here [OK]

Common Mistakes:

• Assuming negative values get green
• Confusing color assignment logic
• Ignoring the else condition in list comprehension

4. The following code is intended to create a diverging bar chart but does not color bars correctly. What is the error?

values = [5, -2, 3, -4]
y_pos = range(len(values))
colors = ['green' if v < 0 else 'red' for v in values]
plt.barh(y_pos, values, color=colors)
plt.show()

medium

A. The color condition is reversed; positive values should be green

B. The y_pos variable should be a list, not a range

C. The barh function cannot accept a color list

D. Values should be absolute for coloring

Solution

1. Step 1: Review the color condition

   The code assigns 'green' if value < 0, else 'red'. This means negative values are green and positive are red.

2. Step 2: Understand diverging bar chart color convention

   Usually, positive values are green and negative values are red to show gains and losses clearly.

3. Final Answer:

   The color condition is reversed; positive values should be green -> Option A

4. Quick Check:

   Positive = green, negative = red [OK]

Hint: Positive bars usually green, negative bars red [OK]

Common Mistakes:

• Reversing color conditions
• Thinking range is invalid for y_pos
• Using absolute values for coloring

5. You have a dataset with sales gains and losses for 5 products: {'A': 10, 'B': -5, 'C': 0, 'D': 7, 'E': -3}. How would you create a diverging bar chart that clearly shows gains in green, losses in red, and zero values in gray using matplotlib?

hard

A. Use a color list with a condition: green if >0, red if <0, gray if 0, then plot with plt.barh

B. Plot absolute values only with a single color to avoid confusion

C. Use plt.barh with colors assigned randomly

D. Plot only positive values and ignore zeros and negatives

Solution

1. Step 1: Define color conditions for all cases

   Assign 'green' for positive, 'red' for negative, and 'gray' for zero values to distinguish all clearly.

2. Step 2: Create color list and plot

   Use a list comprehension to assign colors based on value sign, then plot with plt.barh using these colors.

3. Final Answer:

   Use a color list with a condition: green if >0, red if <0, gray if 0, then plot with plt.barh -> Option A

4. Quick Check:

   Color by sign including zero = Use a color list with a condition: green if >0, red if <0, gray if 0, then plot with plt.barh [OK]

Hint: Add gray for zero values in color list [OK]

Common Mistakes:

• Ignoring zero values color
• Using absolute values losing sign info
• Assigning random colors without logic