Why categorical visualization matters in Matplotlib - Challenge Your Understanding

Challenge - 5 Problems

❓ Predict Output

intermediate

What is the output of this bar chart code?

Look at the code below that creates a bar chart for fruit counts. What will the heights of the bars be?

Matplotlib

import matplotlib.pyplot as plt
fruits = ['Apple', 'Banana', 'Cherry']
counts = [10, 15, 7]
plt.bar(fruits, counts)
plt.show()

A Bars with heights 10, 7, and 15 for Apple, Banana, and Cherry respectively

B Bars with heights 7, 10, and 15 for Apple, Banana, and Cherry respectively

C Bars with heights 15, 10, and 7 for Apple, Banana, and Cherry respectively

D Bars with heights 10, 15, and 7 for Apple, Banana, and Cherry respectively

💡 Hint

Check the order of the counts list and how it matches the fruits list.

The plt.bar function pairs each fruit with the count at the same index. So Apple gets 10, Banana 15, and Cherry 7.

❓ data_output

intermediate

How many categories are shown in this pie chart?

Given this pie chart code, how many slices (categories) will appear?

Matplotlib

import matplotlib.pyplot as plt
labels = ['Red', 'Blue', 'Green', 'Yellow']
sizes = [20, 30, 25, 25]
plt.pie(sizes, labels=labels)
plt.show()

A 2

B 4

C 5

D 3

💡 Hint

Count the number of labels provided.

Each label corresponds to one slice in the pie chart, so there are 4 slices.

❓ visualization

advanced

Which option produces a correct stacked bar chart?

You want to show sales of two products over three months stacked in one bar per month. Which code produces the correct stacked bar chart?

Matplotlib

import matplotlib.pyplot as plt
months = ['Jan', 'Feb', 'Mar']
sales_A = [5, 7, 6]
sales_B = [3, 4, 5]

A

plt.bar(months, sales_A)
plt.bar(months, sales_B)
plt.show()

B

plt.bar(months, sales_A + sales_B)
plt.show()

C

plt.bar(months, sales_A)
plt.bar(months, sales_B, bottom=sales_A)
plt.show()

D

plt.bar(months, sales_B, bottom=sales_A)
plt.bar(months, sales_A)
plt.show()

💡 Hint

Stacked bars need the bottom parameter to stack the second bar on top of the first.

Option C stacks sales_B on top of sales_A using the bottom parameter. Option C tries to add lists which causes an error. Option C overlays bars without stacking. Option C stacks in wrong order causing incorrect visualization.

🧠 Conceptual

advanced

Why is categorical visualization important in data science?

Which option best explains why visualizing categorical data helps in data science?

A It helps to quickly compare groups and see patterns or differences between categories.

B It is only useful for numerical data and does not apply to categories.

C It replaces the need for any statistical analysis.

D It makes data look colorful but does not add understanding.

💡 Hint

Think about how charts help us understand group differences.

Categorical visualization helps us see how different groups compare, making patterns easier to spot.

🔧 Debug

expert

What error does this code raise?

This code tries to plot a bar chart but fails. What error will it raise?

Matplotlib

import matplotlib.pyplot as plt
categories = ['A', 'B', 'C']
values = [10, 20]
plt.bar(categories, values)
plt.show()

A ValueError: x and height must have the same size

B TypeError: unsupported operand type(s) for +: 'list' and 'int'

C SyntaxError: invalid syntax

D IndexError: list index out of range

💡 Hint

Check if the categories and values lists have the same length.

The categories list has 3 items but values has only 2, so matplotlib raises a ValueError about size mismatch.

Practice

1. Why is categorical visualization important in data science?

easy

A. It shows exact numerical calculations only.

B. It helps compare data grouped by categories clearly.

C. It replaces the need for any data cleaning.

D. It automatically predicts future data trends.

Solution

1. Step 1: Understand the purpose of categorical visualization

   Categorical visualization groups data by categories to make comparisons easy and clear.

2. Step 2: Evaluate the options based on this purpose

   Only It helps compare data grouped by categories clearly. correctly states that it helps compare grouped data clearly; others describe unrelated functions.

3. Final Answer:

   It helps compare data grouped by categories clearly. -> Option B

4. Quick Check:

   Categorical visualization = clear category comparison [OK]

Hint: Remember: categorical charts show groups clearly [OK]

Common Mistakes:

• Confusing categorical visualization with prediction
• Thinking it shows exact numbers only
• Assuming it replaces data cleaning

2. Which of the following is the correct way to create a bar chart for categories using matplotlib?

easy

A. plt.bar(categories, values)

B. plt.plot(categories, values)

C. plt.scatter(categories, values)

D. plt.hist(categories, values)

Solution

1. Step 1: Identify the matplotlib function for bar charts

   The plt.bar() function is used to create bar charts for categorical data.

2. Step 2: Check other options for correctness

   plt.plot() is for line plots, plt.scatter() for scatter plots, and plt.hist() for histograms, none are best for categorical bars.

3. Final Answer:

   plt.bar(categories, values) -> Option A

4. Quick Check:

   Bar chart function = plt.bar() [OK]

Hint: Bar charts use plt.bar() in matplotlib [OK]

Common Mistakes:

• Using plt.plot() for categorical bars
• Confusing scatter plots with bar charts
• Using histogram function for categories

3. What will the following code display?

import matplotlib.pyplot as plt
categories = ['A', 'B', 'C']
values = [5, 3, 7]
plt.bar(categories, values)
plt.title('Category Counts')
plt.show()

medium

A. A bar chart with bars for A=5, B=3, C=7 and title 'Category Counts'.

B. A line chart connecting points A=5, B=3, C=7.

C. An error because plt.bar() cannot take lists.

D. A scatter plot with points at (A,5), (B,3), (C,7).

Solution

1. Step 1: Understand plt.bar() with category and value lists

   The code creates a bar chart with categories on x-axis and values as bar heights.

2. Step 2: Confirm the title and display

   The title 'Category Counts' is set and plt.show() displays the chart correctly.

3. Final Answer:

   A bar chart with bars for A=5, B=3, C=7 and title 'Category Counts'. -> Option A

4. Quick Check:

   plt.bar() with lists = bar chart [OK]

Hint: plt.bar() with lists shows bars by category [OK]

Common Mistakes:

• Thinking plt.bar() creates line or scatter plots
• Believing plt.bar() cannot take lists
• Ignoring the title setting

4. Identify the error in this code for categorical visualization:

import matplotlib.pyplot as plt
categories = ['X', 'Y', 'Z']
values = [10, 15]
plt.bar(categories, values)
plt.show()

medium

A. plt.bar() cannot plot string categories.

B. The values list should be strings, not numbers.

C. plt.show() is missing.

D. The categories and values lists have different lengths.

Solution

1. Step 1: Check lengths of categories and values

   Categories list has 3 items, values list has 2 items, causing mismatch error.

2. Step 2: Verify other options

   plt.bar() accepts string categories, plt.show() is present, and values should be numbers for bar heights.

3. Final Answer:

   The categories and values lists have different lengths. -> Option D

4. Quick Check:

   List length mismatch = error [OK]

Hint: Check if categories and values lists match in length [OK]

Common Mistakes:

• Assuming plt.bar() can't use strings
• Forgetting plt.show()
• Using strings for values instead of numbers

5. You have sales data for three products: 'Shoes', 'Hats', and 'Shirts' with sales 120, 80, and 150 respectively. You want to create a bar chart that clearly shows which product sold the most. Which steps should you follow to make the chart easy to understand?

hard

A. Use plt.scatter() with product names as x and sales as y, no title.

B. Use plt.plot() with sales only, no labels needed.

C. Use plt.bar() with product names and sales, add clear labels and a title.

D. Print sales numbers only, no chart needed.

Solution

1. Step 1: Choose the correct plot type for categorical data

   Bar charts (plt.bar()) are best to compare sales by product category clearly.

2. Step 2: Add labels and title for clarity

   Adding x-axis labels (product names), y-axis labels (sales), and a title helps others understand the chart quickly.

3. Final Answer:

   Use plt.bar() with product names and sales, add clear labels and a title. -> Option C

4. Quick Check:

   Bar chart + labels + title = clear visualization [OK]

Hint: Bar chart + labels + title = clear category comparison [OK]

Common Mistakes:

• Using line or scatter plots for categories
• Skipping labels and titles
• Not visualizing data at all