Colorbar formatting in Matplotlib - Mini Project: Build & Apply

Colorbar formatting

📖 Scenario: You are working with a heatmap to visualize temperature data across a city grid. You want to add a colorbar to your heatmap to help viewers understand the temperature scale.

🎯 Goal: Create a heatmap using matplotlib and add a colorbar with customized formatting to clearly show the temperature scale.

📋 What You'll Learn

Create a 2D list called temperature_data with exact values

Create a matplotlib figure and axis

Plot the heatmap using imshow with temperature_data

Add a colorbar to the heatmap and format its label and ticks

Print the colorbar's label text to confirm formatting

💡 Why This Matters

🌍 Real World

Colorbars help people understand the meaning of colors in heatmaps or other visualizations, such as temperature maps or population density maps.

💼 Career

Data scientists and analysts often customize colorbars to make their visualizations clearer and more professional for reports and presentations.

1

Create the temperature data

Create a 2D list called temperature_data with these exact values: [[22, 24, 23], [25, 27, 26], [24, 23, 22]]

Matplotlib

# Create the temperature_data list with the exact values
# Your code here

Hint

Use square brackets to create a list of lists with the exact numbers.

2

Set up the plot

Import matplotlib.pyplot as plt. Create a figure and axis using plt.subplots().

Matplotlib

temperature_data = [[22, 24, 23], [25, 27, 26], [24, 23, 22]]
# Import matplotlib.pyplot as plt and create figure and axis
# Your code here

Hint

Use import matplotlib.pyplot as plt and then fig, ax = plt.subplots().

3

Plot heatmap and add colorbar

Use ax.imshow(temperature_data) to plot the heatmap. Then add a colorbar with fig.colorbar() and assign it to cbar. Set the colorbar label to 'Temperature (°C)' and set the label font size to 12.

Matplotlib

import matplotlib.pyplot as plt

temperature_data = [[22, 24, 23], [25, 27, 26], [24, 23, 22]]

fig, ax = plt.subplots()
# Plot heatmap and add colorbar with label and font size
# Your code here

Hint

Use ax.imshow() to plot, then fig.colorbar() to add the colorbar. Use cbar.set_label() to set the label and font size.

4

Display the plot and print colorbar label

Print the colorbar label text using print(cbar.ax.yaxis.label.get_text()). Then display the plot with plt.show().

Matplotlib

import matplotlib.pyplot as plt

temperature_data = [[22, 24, 23], [25, 27, 26], [24, 23, 22]]

fig, ax = plt.subplots()

heatmap = ax.imshow(temperature_data)
cbar = fig.colorbar(heatmap)
cbar.set_label('Temperature (°C)', fontsize=12)
# Print the colorbar label text and show the plot
# Your code here

Hint

Use print(cbar.ax.yaxis.label.get_text()) to print the label text, then plt.show() to display the plot.

Practice

1. What is the main purpose of adding a colorbar to a matplotlib plot?

easy

A. To show the scale of colors representing data values

B. To add a title to the plot

C. To change the background color of the plot

D. To display the axis labels

Solution

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

   Colors in a plot often represent data values or categories.

2. Step 2: Purpose of colorbar

   A colorbar explains what each color means by showing a scale or legend.

3. Final Answer:

   To show the scale of colors representing data values -> Option A

4. Quick Check:

   Colorbar = color scale legend [OK]

Hint: Colorbar explains colors' meaning in data [OK]

Common Mistakes:

• Confusing colorbar with plot title
• Thinking colorbar changes plot background
• Mixing colorbar with axis labels

2. Which of the following is the correct way to add a colorbar to a matplotlib plot stored in variable im?

easy

A. plt.colorbar()

B. plt.colorbar(im)

C. im.colorbar()

D. plt.add_colorbar(im)

Solution

1. Step 1: Recall matplotlib colorbar syntax

   The function plt.colorbar() requires the image or mappable object as an argument.

2. Step 2: Check each option

   plt.colorbar(im) correctly calls plt.colorbar(im). plt.colorbar() misses the argument. im.colorbar() is invalid method. plt.add_colorbar(im) is not a matplotlib function.

3. Final Answer:

   plt.colorbar(im) -> Option B

4. Quick Check:

   plt.colorbar(im) adds colorbar [OK]

Hint: Always pass the image object to plt.colorbar() [OK]

Common Mistakes:

• Calling plt.colorbar() without argument
• Using non-existent methods like im.colorbar()
• Trying plt.add_colorbar() which doesn't exist

3. What will be the effect of this code snippet on the colorbar ticks?

import matplotlib.pyplot as plt
import numpy as np

img = plt.imshow(np.random.rand(5,5))
cbar = plt.colorbar(img)
cbar.set_ticks([0, 0.5, 1])
cbar.set_ticklabels(['Low', 'Medium', 'High'])
plt.show()

medium

A. Colorbar will have no ticks displayed

B. Colorbar will show default ticks ignoring set_ticks and set_ticklabels

C. Code will raise an error because set_ticklabels is invalid

D. Colorbar will show ticks at 0, 0.5, 1 labeled as Low, Medium, High

Solution

1. Step 1: Understand set_ticks and set_ticklabels

   set_ticks sets positions of ticks; set_ticklabels sets their text labels.

2. Step 2: Analyze the code

   The code sets ticks at 0, 0.5, 1 and labels them as Low, Medium, High respectively.

3. Final Answer:

   Colorbar will show ticks at 0, 0.5, 1 labeled as Low, Medium, High -> Option D

4. Quick Check:

   set_ticks + set_ticklabels changes tick positions and labels [OK]

Hint: Use set_ticks and set_ticklabels to customize colorbar ticks [OK]

Common Mistakes:

• Assuming set_ticklabels is invalid
• Expecting default ticks despite setting new ones
• Thinking ticks will disappear

4. Identify the error in this code that tries to format the colorbar ticks:

import matplotlib.pyplot as plt
import numpy as np

img = plt.imshow(np.random.rand(5,5))
cbar = plt.colorbar(img)
cbar.set_ticks([0, 0.5, 1])
cbar.set_ticklabels('Low', 'Medium', 'High')
plt.show()

medium

A. plt.colorbar() cannot be assigned to a variable

B. set_ticks must be called after set_ticklabels

C. set_ticklabels expects a list, not separate string arguments

D. imshow requires integer data, not float

Solution

1. Step 1: Check set_ticklabels usage

   set_ticklabels requires a single list or iterable of labels, not separate arguments.

2. Step 2: Identify the error in code

   The code passes separate strings instead of a list, causing a TypeError.

3. Final Answer:

   set_ticklabels expects a list, not separate string arguments -> Option C

4. Quick Check:

   set_ticklabels(['Low', 'Medium', 'High']) correct [OK]

Hint: Pass labels as a list to set_ticklabels [OK]

Common Mistakes:

• Passing multiple strings instead of a list
• Calling set_ticks after set_ticklabels (order doesn't matter)
• Misunderstanding plt.colorbar() return value
• Thinking imshow needs integers only

5. You want to format a colorbar to show percentage ticks from 0% to 100% with steps of 25%. Which code snippet correctly formats the colorbar ticks and labels?

hard

A. cbar.set_ticks([0, 0.25, 0.5, 0.75, 1])
cbar.set_ticklabels(['0%', '25%', '50%', '75%', '100%'])

B. cbar.set_ticks(['0%', '25%', '50%', '75%', '100%'])
cbar.set_ticklabels([0, 0.25, 0.5, 0.75, 1])

C. cbar.set_ticks([0, 25, 50, 75, 100])
cbar.set_ticklabels(['0%', '25%', '50%', '75%', '100%'])

D. cbar.set_ticks([0, 0.25, 0.5, 0.75, 1])
cbar.set_ticklabels([0, 25, 50, 75, 100])

Solution

1. Step 1: Understand colorbar data range

   Colorbar values range from 0 to 1 for normalized data.

2. Step 2: Set ticks and labels correctly

   Ticks must be numeric positions (0 to 1), labels are strings with % signs.

3. Step 3: Analyze options

   cbar.set_ticks([0, 0.25, 0.5, 0.75, 1]) cbar.set_ticklabels(['0%', '25%', '50%', '75%', '100%']) correctly sets numeric ticks and string labels. cbar.set_ticks(['0%', '25%', '50%', '75%', '100%']) cbar.set_ticklabels([0, 0.25, 0.5, 0.75, 1]) swaps types incorrectly. cbar.set_ticks([0, 25, 50, 75, 100]) cbar.set_ticklabels(['0%', '25%', '50%', '75%', '100%']) uses ticks outside 0-1 range. cbar.set_ticks([0, 0.25, 0.5, 0.75, 1]) cbar.set_ticklabels([0, 25, 50, 75, 100]) sets numeric ticks but labels as numbers without %.

4. Final Answer:

   cbar.set_ticks([0, 0.25, 0.5, 0.75, 1]) cbar.set_ticklabels(['0%', '25%', '50%', '75%', '100%']) -> Option A

5. Quick Check:

   Ticks numeric, labels string with % [OK]

Hint: Ticks numeric 0-1; labels as strings with % [OK]

Common Mistakes:

• Using tick positions as strings
• Setting ticks outside 0-1 range
• Labeling ticks with numbers missing % sign