Heatmap with plt.imshow in Matplotlib - Interactive Code Practice

Practice - 5 Tasks

1fill in blank

easy

Complete the code to display the heatmap using plt.imshow.

Matplotlib

import matplotlib.pyplot as plt
import numpy as np

data = np.array([[1, 2], [3, 4]])
plt.[1](data)
plt.show()

Drag options to blanks, or click blank then click option'

Ascatter

Bplot

Cimshow

Dbar

Attempts:

3 left

💡 Hint

Common Mistakes

Using plt.plot() which is for line plots.

Using plt.scatter() which is for scatter plots.

plt.imshow() is used to display data as an image, which is perfect for heatmaps.

2fill in blank

medium

Complete the code to add a colorbar to the heatmap.

Matplotlib

import matplotlib.pyplot as plt
import numpy as np

data = np.array([[5, 10], [15, 20]])
plt.imshow(data)
plt.[1]()
plt.show()

Drag options to blanks, or click blank then click option'

Alegend

Bcolorbar

Ctitle

Dxlabel

Attempts:

3 left

💡 Hint

Common Mistakes

Using plt.legend() which is for plot legends.

Trying to use plt.title() to add color scale.

plt.colorbar() adds a color scale bar to the heatmap to show value mapping.

3fill in blank

hard

Fix the error in the code to correctly display the heatmap with a 'hot' colormap.

Matplotlib

import matplotlib.pyplot as plt
import numpy as np

data = np.array([[7, 8], [9, 10]])
plt.imshow(data, cmap=[1])
plt.colorbar()
plt.show()

Drag options to blanks, or click blank then click option'

A'cool'

B'gray'

C'spring'

D'hot'

Attempts:

3 left

💡 Hint

Common Mistakes

Forgetting to put the colormap name in quotes.

Using a colormap that doesn't highlight heat well.

The 'hot' colormap shows colors from black to red to yellow to white, good for heatmaps.

4fill in blank

hard

Fill both blanks to create a heatmap with data normalized between 0 and 1 and use the 'viridis' colormap.

Matplotlib

import matplotlib.pyplot as plt
import numpy as np

data = np.array([[10, 20], [30, 40]])
norm_data = data / [1]
plt.imshow(norm_data, cmap=[2])
plt.colorbar()
plt.show()

Drag options to blanks, or click blank then click option'

A40

B30

C'viridis'

D'plasma'

Attempts:

3 left

💡 Hint

Common Mistakes

Dividing by a wrong number that is not the max value.

Using a colormap name without quotes.

Dividing by 40 normalizes data to max 1; 'viridis' is a popular colormap for heatmaps.

5fill in blank

hard

Fill all three blanks to create a heatmap from a 3x3 matrix, set interpolation to 'nearest', and add a title 'My Heatmap'.

Matplotlib

import matplotlib.pyplot as plt
import numpy as np

data = np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]])
plt.imshow(data, interpolation=[1], cmap=[2])
plt.colorbar()
plt.title([3])
plt.show()

Drag options to blanks, or click blank then click option'

A'bilinear'

B'nearest'

C'viridis'

D'My Heatmap'

Attempts:

3 left

💡 Hint

Common Mistakes

Using interpolation values without quotes.

Forgetting quotes around the title string.

Interpolation 'nearest' shows sharp blocks; 'viridis' is a good colormap; title is a string.

Practice

1. What does the plt.imshow function do in matplotlib when given a 2D array?

easy

A. It prints the array values as text on the plot.

B. It plots a line graph of the array values.

C. It creates a heatmap by showing the array values as colors.

D. It creates a pie chart from the array values.

Solution

1. Step 1: Understand plt.imshow input

   The function takes a 2D array and maps its values to colors in a grid.

2. Step 2: Recognize heatmap behavior

   This color mapping creates a heatmap, showing patterns visually.

3. Final Answer:

   It creates a heatmap by showing the array values as colors. -> Option C

4. Quick Check:

   plt.imshow = heatmap colors [OK]

Hint: plt.imshow shows 2D data as colored grids [OK]

Common Mistakes:

• Thinking plt.imshow plots lines or text
• Confusing heatmap with pie chart
• Assuming plt.imshow prints numbers

2. Which of these is the correct way to add a color bar to a heatmap created with plt.imshow?

easy

A. plt.colorbar()

B. plt.addcolor()

C. plt.showcolor()

D. plt.colorheat()

Solution

1. Step 1: Recall color bar function

   Matplotlib uses plt.colorbar() to add a color scale next to images.

2. Step 2: Check other options

   Other options are not valid matplotlib functions.

3. Final Answer:

   plt.colorbar() -> Option A

4. Quick Check:

   Color bar function = plt.colorbar() [OK]

Hint: Use plt.colorbar() to add color scale [OK]

Common Mistakes:

• Using non-existent functions like plt.addcolor()
• Forgetting to call plt.colorbar() after plt.imshow()
• Trying to add color bar before plotting heatmap

3. What will the following code display?

import matplotlib.pyplot as plt
import numpy as np

data = np.array([[1, 2], [3, 4]])
plt.imshow(data, cmap='hot')
plt.colorbar()
plt.show()

medium

A. A scatter plot of points (1,2) and (3,4) with hot colors.

B. A line plot of the numbers 1, 2, 3, 4 with a color bar.

C. An error because cmap='hot' is invalid.

D. A 2x2 heatmap with colors from black to red to yellow, and a color bar showing values 1 to 4.

Solution

1. Step 1: Understand plt.imshow with cmap='hot'

   The data array is 2x2, so plt.imshow creates a 2x2 grid colored by values using the 'hot' color map (black to red to yellow).

2. Step 2: Effect of plt.colorbar()

   The color bar shows the scale from the minimum value 1 to maximum 4, helping interpret colors.

3. Final Answer:

   A 2x2 heatmap with colors from black to red to yellow, and a color bar showing values 1 to 4. -> Option D

4. Quick Check:

   plt.imshow + cmap='hot' + colorbar = heatmap with hot colors [OK]

Hint: cmap='hot' colors low to high values black-red-yellow [OK]

Common Mistakes:

• Thinking plt.imshow plots lines or scatter
• Believing cmap='hot' causes error
• Ignoring color bar meaning

4. The code below is intended to show a heatmap but does not display the color bar. What is the error?

import matplotlib.pyplot as plt
import numpy as np

data = np.random.rand(3,3)
plt.imshow(data)
plt.colorbar
plt.show()

medium

A. plt.colorbar is missing parentheses, so color bar is not shown.

B. plt.imshow needs a cmap argument to show color bar.

C. plt.show() must come before plt.colorbar().

D. np.random.rand(3,3) creates invalid data for heatmap.

Solution

1. Step 1: Identify plt.colorbar usage

   The code uses plt.colorbar without parentheses, so the function is not called.

2. Step 2: Correct function call

   Adding parentheses plt.colorbar() calls the function and displays the color bar.

3. Final Answer:

   plt.colorbar is missing parentheses, so color bar is not shown. -> Option A

4. Quick Check:

   Call plt.colorbar() with () to show color bar [OK]

Hint: Always call plt.colorbar() with parentheses [OK]

Common Mistakes:

• Forgetting parentheses on plt.colorbar
• Thinking cmap is required for color bar
• Calling plt.show() before plt.colorbar

5. You have a 4x4 matrix with values from 0 to 15. You want to create a heatmap using plt.imshow that shows the values clearly with a color bar, but you want the color scale to always be from 0 to 20 regardless of data. Which code snippet correctly does this?

hard

A. plt.imshow(matrix, vmin=0, vmax=15)
plt.colorbar()
plt.show()

B. plt.imshow(matrix, vmin=0, vmax=20)
plt.colorbar()
plt.show()

C. plt.imshow(matrix, cmap='gray')
plt.colorbar()
plt.show()

D. plt.imshow(matrix)
plt.colorbar(vmin=0, vmax=20)
plt.show()

Solution

1. Step 1: Set color scale limits in plt.imshow

   Use vmin=0 and vmax=20 in plt.imshow to fix color scale from 0 to 20.

2. Step 2: Add color bar normally

   Call plt.colorbar() to show the color scale matching the fixed limits.

3. Final Answer:

   plt.imshow(matrix, vmin=0, vmax=20) plt.colorbar() plt.show() -> Option B

4. Quick Check:

   Set vmin/vmax in imshow, then add colorbar [OK]

Hint: Use vmin and vmax in plt.imshow to fix color scale [OK]

Common Mistakes:

• Setting vmin/vmax in plt.colorbar instead of plt.imshow
• Not setting vmax to 20, so scale auto adjusts
• Using wrong vmin/vmax values