Why color matters in visualization in Matplotlib - See It in Action

Why Color Matters in Visualization

📖 Scenario: You are working as a data analyst for a local grocery store. You want to show the sales of different fruit types using a bar chart. You will learn how color can help make your chart easier to understand and more attractive.

🎯 Goal: Create a bar chart showing fruit sales with colors that help viewers quickly see which fruit sold the most and which sold the least.

📋 What You'll Learn

Create a dictionary with fruit names and their sales numbers.

Create a list of colors to use for each fruit bar.

Use matplotlib to plot a bar chart with the fruit names on the x-axis and sales on the y-axis.

Apply the colors list to the bars to highlight differences clearly.

Print the sales dictionary at the end.

💡 Why This Matters

🌍 Real World

Colors in charts help customers and managers quickly see important information, like which products sell best.

💼 Career

Data analysts and scientists use color in visualizations to make reports easier to understand and more persuasive.

1

Create the fruit sales data

Create a dictionary called fruit_sales with these exact entries: 'Apples': 50, 'Bananas': 30, 'Cherries': 20, 'Dates': 15, 'Elderberries': 5.

Matplotlib

# Create the fruit_sales dictionary with exact values
# Your code here

Hint

Use curly braces to create a dictionary. Each fruit name is a key and its sales number is the value.

2

Create a list of colors for the bars

Create a list called colors with these exact color names in order: 'red', 'yellow', 'pink', 'brown', 'purple'.

Matplotlib

fruit_sales = {'Apples': 50, 'Bananas': 30, 'Cherries': 20, 'Dates': 15, 'Elderberries': 5}
# Create the colors list with exact color names
# Your code here

Hint

Use square brackets to create a list. Put the color names as strings in the order matching the fruits.

3

Plot the bar chart with colors

Use matplotlib to plot a bar chart with fruit names on the x-axis and sales numbers on the y-axis. Use the colors list to color each bar. Use plt.bar(fruit_sales.keys(), fruit_sales.values(), color=colors).

Matplotlib

import matplotlib.pyplot as plt

fruit_sales = {'Apples': 50, 'Bananas': 30, 'Cherries': 20, 'Dates': 15, 'Elderberries': 5}
colors = ['red', 'yellow', 'pink', 'brown', 'purple']
# Plot the bar chart using plt.bar with fruit_sales keys and values and colors
# Your code here

Hint

Use plt.bar() with the keys and values of the dictionary and the colors list for the color argument.

4

Show the plot and print the sales data

Add plt.show() to display the chart. Then print the fruit_sales dictionary using print(fruit_sales).

Matplotlib

import matplotlib.pyplot as plt

fruit_sales = {'Apples': 50, 'Bananas': 30, 'Cherries': 20, 'Dates': 15, 'Elderberries': 5}
colors = ['red', 'yellow', 'pink', 'brown', 'purple']
plt.bar(fruit_sales.keys(), fruit_sales.values(), color=colors)
plt.xlabel('Fruit')
plt.ylabel('Sales')
plt.title('Fruit Sales with Color Highlight')
# Show the plot and print the fruit_sales dictionary
# Your code here

Hint

Use plt.show() to display the chart. Then use print(fruit_sales) to show the dictionary.

Practice

1.

Why is color important in data visualization?

easy

A. It helps to group and highlight important data points.

B. It makes the plot run faster.

C. It reduces the size of the data file.

D. It changes the data values automatically.

Solution

1. Step 1: Understand the role of color in visualization

   Color is used to make data easier to understand by grouping or highlighting.

2. Step 2: Evaluate other options

   Options A, B, and D describe effects unrelated to color's purpose in visualization.

3. Final Answer:

   It helps to group and highlight important data points. -> Option A

4. Quick Check:

   Color improves understanding = C [OK]

Hint: Color groups and highlights data clearly [OK]

Common Mistakes:

• Thinking color speeds up plotting
• Believing color changes data values
• Confusing color with file size

2.

Which of the following matplotlib code snippets correctly sets the color of a line plot to red?

easy

A. plt.plot(x, y, col='red')

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

C. plt.plot(x, y, c='blue')

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

Solution

1. Step 1: Check correct parameter for color in plt.plot()

   The correct parameter is color with a string value for color name.

2. Step 2: Validate each option

   plt.plot(x, y, color='red') uses color='red' correctly. plt.plot(x, y, col='red') uses wrong parameter col. plt.plot(x, y, c='blue') uses c='blue' which is valid but sets blue, not red. plt.plot(x, y, color=red) misses quotes around 'red'.

3. Final Answer:

   plt.plot(x, y, color='red') -> Option B

4. Quick Check:

   Correct color param = B [OK]

Hint: Use 'color' param with quotes for color name [OK]

Common Mistakes:

• Using 'col' instead of 'color'
• Forgetting quotes around color name
• Confusing 'c' param color with wrong color

3.

What will be the color of the bars in this code?

import matplotlib.pyplot as plt
values = [3, 7, 5]
plt.bar([1, 2, 3], values, color=['green', 'red', 'blue'])
plt.show()

medium

A. Bars will be all red.

B. All bars will be green.

C. Bars will be green, red, and blue respectively.

D. Code will raise an error due to color list.

Solution

1. Step 1: Understand color parameter with list in plt.bar()

   When color is a list, each bar gets the corresponding color in order.

2. Step 2: Match colors to bars

   Bars at positions 1, 2, 3 get colors green, red, blue respectively.

3. Final Answer:

   Bars will be green, red, and blue respectively. -> Option C

4. Quick Check:

   List colors assign per bar = A [OK]

Hint: List colors assign to bars in order [OK]

Common Mistakes:

• Assuming all bars get first color
• Expecting error from list of colors
• Confusing color param with single string only

4.

Find the error in this code that tries to color a scatter plot points blue:

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

medium

A. The color value blue is not in quotes.

B. The scatter function does not accept color parameter.

C. x and y lists must be numpy arrays.

D. plt.show() is missing.

Solution

1. Step 1: Check color parameter usage

   Color names must be strings, so 'blue' needs quotes.

2. Step 2: Verify other parts of code

   Scatter accepts color param, x and y can be lists, plt.show() is present.

3. Final Answer:

   The color value blue is not in quotes. -> Option A

4. Quick Check:

   Color names need quotes = D [OK]

Hint: Put color names inside quotes [OK]

Common Mistakes:

• Forgetting quotes around color names
• Thinking scatter lacks color param
• Assuming lists can't be used for x,y

5.

You want to create a line plot with two lines: one for sales in 2023 and one for 2024. You want to use colors that are easy to distinguish for colorblind viewers. Which color pair is best to use?

• A) Blue and Orange
• B) Red and Green
• C) Purple and Pink
• D) Yellow and Light Green

hard

A. Yellow and Light Green

B. Red and Green

C. Purple and Pink

D. Blue and Orange

Solution

1. Step 1: Understand colorblind-friendly colors

   Red and green are hard to distinguish for many colorblind people. Blue and orange are easier to tell apart.

2. Step 2: Evaluate color pairs for clarity

   Purple and pink or yellow and light green can be too similar or low contrast. Blue and orange provide clear contrast.

3. Final Answer:

   Blue and Orange -> Option D

4. Quick Check:

   Colorblind-friendly pairs = A [OK]

Hint: Use blue and orange for colorblind-friendly contrast [OK]

Common Mistakes:

• Choosing red and green which confuse colorblind users
• Picking colors with low contrast
• Ignoring accessibility in color choice