Line colors and width in Matplotlib - Time & Space Complexity

Time Complexity: Line colors and width

O(n)

Understanding Time Complexity

We want to understand how the time to draw lines changes when we change their colors and widths in matplotlib.

How does the drawing time grow as we add more lines with different colors and widths?

Scenario Under Consideration

Analyze the time complexity of the following code snippet.

import matplotlib.pyplot as plt

x = range(1000)
for i in range(100):
    plt.plot(x, [j + i for j in x], color=f'C{i%10}', linewidth=0.5 + i*0.05)
plt.show()

This code draws 100 lines, each with a different color and width, on the same plot.

Identify Repeating Operations

Identify the loops, recursion, array traversals that repeat.

• Primary operation: The loop that draws each line with specific color and width.
• How many times: 100 times, once per line.

How Execution Grows With Input

As the number of lines increases, the drawing time grows roughly in direct proportion.

Input Size (n)	Approx. Operations
10	10 line draws
100	100 line draws
1000	1000 line draws

Pattern observation: Doubling the number of lines roughly doubles the work done.

Final Time Complexity

Time Complexity: O(n)

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

Common Mistake

[X] Wrong: "Changing line colors or widths makes the drawing time grow faster than the number of lines."

[OK] Correct: The color and width settings are simple properties applied per line and do not add extra loops or nested work, so they do not increase the growth rate beyond linear.

Interview Connect

Understanding how drawing operations scale helps you reason about performance in data visualization tasks, a useful skill in many data science projects.

Self-Check

What if we added nested loops to draw multiple points per line with different styles? How would the time complexity change?

Practice

1. Which parameter in matplotlib.pyplot.plot() changes the color of the line?

easy

A. linestyle

B. color

C. linewidth

D. marker

Solution

1. Step 1: Understand the purpose of parameters in plot()

   The color parameter sets the line color, while linewidth changes thickness.

2. Step 2: Identify the parameter for color

   Among the options, only color directly changes the line color.

3. Final Answer:

   color -> Option B

4. Quick Check:

   Line color = color [OK]

Hint: Remember: color changes color, linewidth changes thickness [OK]

Common Mistakes:

• Confusing linewidth with color
• Using linestyle to change color
• Trying to use marker for color

2. Which of the following is the correct syntax to set a red line with thickness 3 in matplotlib.pyplot.plot()?

easy

A. plt.plot(x, y, color='red', linewidth=3)

B. plt.plot(x, y, linewidth='red', color=3)

C. plt.plot(x, y, color=3, linewidth='red')

D. plt.plot(x, y, width=3, color='red')

Solution

1. Step 1: Check parameter names and values

   The correct parameters are color with a color string and linewidth with a number.

2. Step 2: Validate plt.plot(x, y, color='red', linewidth=3) syntax

   plt.plot(x, y, color='red', linewidth=3) uses color='red' and linewidth=3, which is correct syntax.

3. Final Answer:

   plt.plot(x, y, color='red', linewidth=3) -> Option A

4. Quick Check:

   Correct parameter names and types = plt.plot(x, y, color='red', linewidth=3) [OK]

Hint: Use color='colorname' and linewidth=number [OK]

Common Mistakes:

• Swapping color and linewidth values
• Using wrong parameter names like width
• Passing numbers as color

3. What will be the color and thickness of the line drawn by this code?

import matplotlib.pyplot as plt
x = [1, 2, 3]
y = [4, 5, 6]
plt.plot(x, y, color='green', linewidth=5)
plt.show()

medium

A. Blue line with thickness 5

B. Green line with thickness 1

C. Green line with thickness 5

D. Red line with thickness 5

Solution

1. Step 1: Analyze color parameter

   The code sets color='green', so the line color is green.

2. Step 2: Analyze linewidth parameter

   The code sets linewidth=5, so the line thickness is 5 (thicker than default).

3. Final Answer:

   Green line with thickness 5 -> Option C

4. Quick Check:

   color='green' and linewidth=5 = Green line with thickness 5 [OK]

Hint: Check color and linewidth values directly in plot() [OK]

Common Mistakes:

• Ignoring linewidth value
• Assuming default thickness
• Confusing color names

4. Identify the error in this code snippet:

import matplotlib.pyplot as plt
plt.plot([1,2,3], [4,5,6], color=5, linewidth='thick')
plt.show()

medium

A. Missing plt.show() call

B. color should be a number, linewidth should be a string

C. color and linewidth parameters are correct

D. color should be a string, linewidth should be a number

Solution

1. Step 1: Check color parameter type

   color must be a color name string or code, not a number like 5.

2. Step 2: Check linewidth parameter type

   linewidth must be a number (int or float), not a string like 'thick'.

3. Final Answer:

   color should be a string, linewidth should be a number -> Option D

4. Quick Check:

   Parameter types must match expected types [OK]

Hint: color='name', linewidth=number; no numbers for color [OK]

Common Mistakes:

• Passing numbers as color
• Passing strings as linewidth
• Ignoring type errors

5. You want to plot two lines on the same graph: one thick blue line and one thin orange line. Which code correctly does this?

hard

A. plt.plot(x, y1, color='blue', linewidth=6)
plt.plot(x, y2, color='orange', linewidth=1)

B. plt.plot(x, y1, color='orange', linewidth=6)
plt.plot(x, y2, color='blue', linewidth=1)

C. plt.plot(x, y1, linewidth='thick', color='blue')
plt.plot(x, y2, linewidth='thin', color='orange')

D. plt.plot(x, y1, color=6, linewidth='blue')
plt.plot(x, y2, color=1, linewidth='orange')

Solution

1. Step 1: Match colors to lines

   The thick line should be blue, so color='blue' with large linewidth for y1.

2. Step 2: Match thickness and color for second line

   The thin line should be orange, so color='orange' with small linewidth for y2.

3. Step 3: Validate parameter types

   plt.plot(x, y1, color='blue', linewidth=6) plt.plot(x, y2, color='orange', linewidth=1) uses correct parameter names and types for both lines.

4. Final Answer:

   plt.plot(x, y1, color='blue', linewidth=6) plt.plot(x, y2, color='orange', linewidth=1) -> Option A

5. Quick Check:

   Correct colors and linewidths = plt.plot(x, y1, color='blue', linewidth=6) plt.plot(x, y2, color='orange', linewidth=1) [OK]

Hint: Assign color and linewidth clearly for each line [OK]

Common Mistakes:

• Swapping colors and linewidths
• Using strings for linewidth
• Passing numbers as colors