Creating MultiIndex DataFrames in Pandas - Try It Yourself

Creating MultiIndex DataFrames

📖 Scenario: You work in a company that tracks sales data across different regions and product categories. You want to organize this data in a clear way so you can analyze sales by region and category easily.

🎯 Goal: Build a pandas DataFrame with a MultiIndex using region and category as the two levels of the index.

📋 What You'll Learn

Create a pandas DataFrame with sales data for different regions and categories

Use a MultiIndex with two levels: 'Region' and 'Category'

Display the final DataFrame with the MultiIndex

💡 Why This Matters

🌍 Real World

MultiIndex DataFrames help organize complex data with multiple categories, like sales by region and product type, making analysis easier.

💼 Career

Data scientists and analysts often use MultiIndex DataFrames to handle hierarchical data and perform detailed group analysis.

1

Create the sales data dictionary

Create a dictionary called sales_data with these exact keys and values: 'Region' with the list ['North', 'North', 'South', 'South'], 'Category' with the list ['Electronics', 'Furniture', 'Electronics', 'Furniture'], and 'Sales' with the list [2500, 1500, 2000, 1200].

Pandas

# Create the sales_data dictionary with Region, Category, and Sales lists
# Your code here

Hint

Use a dictionary with keys 'Region', 'Category', and 'Sales'. Each key should have a list of values.

2

Create the DataFrame from the dictionary

Create a pandas DataFrame called df from the sales_data dictionary.

Pandas

sales_data = {
    'Region': ['North', 'North', 'South', 'South'],
    'Category': ['Electronics', 'Furniture', 'Electronics', 'Furniture'],
    'Sales': [2500, 1500, 2000, 1200]
}

# Create the DataFrame df from sales_data
# Your code here

Hint

Use pd.DataFrame() with sales_data as the argument.

3

Set MultiIndex using Region and Category

Use df.set_index() with the list ['Region', 'Category'] and inplace=True to set a MultiIndex on df.

Pandas

import pandas as pd

sales_data = {
    'Region': ['North', 'North', 'South', 'South'],
    'Category': ['Electronics', 'Furniture', 'Electronics', 'Furniture'],
    'Sales': [2500, 1500, 2000, 1200]
}

df = pd.DataFrame(sales_data)

# Set MultiIndex on df using Region and Category
# Your code here

Hint

Use df.set_index(['Region', 'Category'], inplace=True) to create the MultiIndex.

4

Display the MultiIndex DataFrame

Write a print() statement to display the DataFrame df with the MultiIndex.

Pandas

import pandas as pd

sales_data = {
    'Region': ['North', 'North', 'South', 'South'],
    'Category': ['Electronics', 'Furniture', 'Electronics', 'Furniture'],
    'Sales': [2500, 1500, 2000, 1200]
}

df = pd.DataFrame(sales_data)

df.set_index(['Region', 'Category'], inplace=True)

# Print the DataFrame df
# Your code here

Hint

Use print(df) to show the DataFrame with the MultiIndex.

Practice

1. What is the main purpose of a MultiIndex DataFrame in pandas?

easy

A. To organize data with multiple levels of row or column labels

B. To store only numeric data efficiently

C. To create plots directly from data

D. To convert DataFrames into lists

Solution

1. Step 1: Understand MultiIndex concept

   A MultiIndex DataFrame allows multiple levels of labels on rows or columns, helping organize complex data.

2. Step 2: Compare options

   Options A, C, and D describe unrelated features: numeric storage, plotting, and conversion, not MultiIndex purpose.

3. Final Answer:

   To organize data with multiple levels of row or column labels -> Option A

4. Quick Check:

   MultiIndex = multiple label levels [OK]

Hint: MultiIndex means multiple label levels in rows or columns [OK]

Common Mistakes:

• Confusing MultiIndex with data type storage
• Thinking MultiIndex is for plotting
• Believing MultiIndex converts DataFrames to lists

2. Which of the following is the correct way to create a MultiIndex from tuples in pandas?

easy

A. pd.MultiIndex.from_tuples([('A', 1), ('B', 2)])

B. pd.MultiIndex.create_tuples([('A', 1), ('B', 2)])

C. pd.MultiIndex.tuples([('A', 1), ('B', 2)])

D. pd.MultiIndex.make_tuples([('A', 1), ('B', 2)])

Solution

1. Step 1: Recall pandas MultiIndex creation methods

   The correct method to create a MultiIndex from tuples is pd.MultiIndex.from_tuples().

2. Step 2: Check method names

   Options A, B, and C use incorrect method names that do not exist in pandas.

3. Final Answer:

   pd.MultiIndex.from_tuples([('A', 1), ('B', 2)]) -> Option A

4. Quick Check:

   Use from_tuples() to create MultiIndex from tuples [OK]

Hint: Use from_tuples() method to create MultiIndex from tuple list [OK]

Common Mistakes:

• Using non-existent methods like create_tuples or make_tuples
• Confusing from_tuples with from_product
• Passing wrong data types instead of list of tuples

3. What will be the output of the following code?

import pandas as pd
index = pd.MultiIndex.from_product([['A', 'B'], [1, 2]])
df = pd.DataFrame({'Value': [10, 20, 30, 40]}, index=index)
df

medium

A. An error because from_product requires numeric inputs only

B. A DataFrame with a single-level index of ['A', 'B', 1, 2] and values 10, 20, 30, 40

C. A DataFrame with columns 'A', 'B', 1, 2 and values 10, 20, 30, 40

D. A DataFrame with index levels 'A' and 'B' combined with 1 and 2, showing values 10, 20, 30, 40

Solution

1. Step 1: Understand pd.MultiIndex.from_product

   This method creates a MultiIndex from the Cartesian product of input iterables, here ['A', 'B'] and [1, 2].

2. Step 2: Analyze DataFrame creation

   The DataFrame uses this MultiIndex as its index and assigns values [10, 20, 30, 40] accordingly, matching the 4 index combinations.

3. Final Answer:

   A DataFrame with index levels 'A' and 'B' combined with 1 and 2, showing values 10, 20, 30, 40 -> Option D

4. Quick Check:

   from_product creates MultiIndex from all pairs [OK]

Hint: from_product creates all pairs of index levels [OK]

Common Mistakes:

• Thinking from_product only accepts numbers
• Confusing MultiIndex with columns
• Expecting a single-level index instead of MultiIndex

4. Identify the error in this code snippet that tries to create a MultiIndex DataFrame:

import pandas as pd
index = pd.MultiIndex.from_tuples(['A', 1], ['B', 2])
df = pd.DataFrame({'Value': [100, 200]}, index=index)
print(df)

medium

A. MultiIndex cannot be used as DataFrame index

B. DataFrame values list length does not match index length

C. from_tuples expects a single list of tuples, not two separate lists

D. pd.DataFrame requires columns parameter when using MultiIndex

Solution

1. Step 1: Check MultiIndex.from_tuples usage

   The method expects one argument: a list of tuples like [('A', 1), ('B', 2)]. The code incorrectly passes two separate lists.

2. Step 2: Verify DataFrame creation

   Because the MultiIndex creation fails, the DataFrame cannot be created properly. Other options are incorrect because the values list length matches the intended index length, MultiIndex can be used as index, and columns parameter is optional.

3. Final Answer:

   from_tuples expects a single list of tuples, not two separate lists -> Option C

4. Quick Check:

   from_tuples needs one list of tuples [OK]

Hint: Pass one list of tuples to from_tuples() [OK]

Common Mistakes:

• Passing multiple lists instead of one list of tuples
• Assuming MultiIndex can't be DataFrame index
• Forgetting to match values length with index length

5. You have sales data for two stores ('Store1', 'Store2') over two years (2022, 2023). How do you create a MultiIndex DataFrame with these as row indices and columns 'Sales' and 'Profit' filled with zeros?

hard

A. Use pd.MultiIndex.from_product([['Sales', 'Profit']], [['Store1', 'Store2'], [2022, 2023]]) as columns and zeros as data

B. Use pd.MultiIndex.from_product([['Store1', 'Store2'], [2022, 2023]]) as index and pd.DataFrame with zeros for columns ['Sales', 'Profit']

C. Use pd.MultiIndex.from_tuples([('Store1', 2022), ('Store2', 2023)]) as index and pd.DataFrame with zeros for columns ['Sales', 'Profit']

D. Create a DataFrame with columns ['Store1', 'Store2', 2022, 2023, 'Sales', 'Profit'] filled with zeros

Solution

1. Step 1: Create MultiIndex for rows

   Use pd.MultiIndex.from_product with store names and years to get all combinations as row indices.

2. Step 2: Create DataFrame with zeros and correct columns

   Use the MultiIndex as index and specify columns ['Sales', 'Profit'], filling data with zeros matching the index length.

3. Final Answer:

   Use pd.MultiIndex.from_product([['Store1', 'Store2'], [2022, 2023]]) as index and pd.DataFrame with zeros for columns ['Sales', 'Profit'] -> Option B

4. Quick Check:

   from_product for all index combos + zeros DataFrame [OK]

Hint: Use from_product for all index combos, zeros for data [OK]

Common Mistakes:

• Using from_tuples with incomplete pairs
• Confusing columns with index levels
• Creating flat DataFrame without MultiIndex