Why DataFrame creation matters in Pandas - Challenge Your Understanding

Challenge - 5 Problems

❓ Predict Output

intermediate

What is the output shape of this DataFrame?

Consider the following code creating a DataFrame from a dictionary of lists. What is the shape (rows, columns) of the resulting DataFrame?

Pandas

import pandas as pd
data = {'A': [1, 2, 3], 'B': [4, 5, 6], 'C': [7, 8, 9]}
df = pd.DataFrame(data)
print(df.shape)

A (2, 3)

B (3, 2)

C (3, 3)

D (9, 3)

💡 Hint

Each key in the dictionary becomes a column, and each list item becomes a row.

The DataFrame has 3 columns (A, B, C) and 3 rows because each list has 3 elements.

❓ data_output

intermediate

What is the DataFrame content after creation?

Given this code, what is the content of the DataFrame?

Pandas

import pandas as pd
data = {'Name': ['Alice', 'Bob'], 'Age': [25, 30]}
df = pd.DataFrame(data)
print(df)

A

Name  Age
1  Alice  25
2  Bob    30

B

   Name  Age
0  Alice   25
1    Bob   30

C

   Name  Age
1  Alice   25
2    Bob   30

D

Name  Age
0  Alice  25
1  Bob    30

💡 Hint

Default index starts at 0 and increments by 1.

The DataFrame shows two rows with index 0 and 1, columns 'Name' and 'Age'.

🔧 Debug

advanced

Why does this DataFrame creation raise an error?

Look at this code snippet. Why does it raise an error?

Pandas

import pandas as pd
data = {'A': [1, 2], 'B': [3, 4, 5]}
df = pd.DataFrame(data)

A ValueError because columns have different length lists

B SyntaxError due to missing colon

C TypeError because dictionary keys are not strings

D No error, DataFrame created successfully

💡 Hint

All columns must have the same number of elements.

The lists for columns 'A' and 'B' have different lengths (2 and 3), causing a ValueError.

🧠 Conceptual

advanced

Why is DataFrame creation method important for performance?

Which statement best explains why the method of creating a DataFrame matters for performance?

A DataFrame creation speed depends only on the number of columns, not rows.

B All DataFrame creation methods have the same performance because pandas optimizes automatically.

C Creating a DataFrame from a CSV file is always faster than from any Python data structure.

D Creating a DataFrame from a list of dictionaries is slower than from a dictionary of lists because of internal data alignment.

💡 Hint

Think about how pandas aligns data internally when creating from different structures.

Creating from a list of dicts requires pandas to align keys for each row, which is slower than creating from dict of lists where columns are aligned directly.

🚀 Application

expert

Which code produces a DataFrame with a MultiIndex?

Choose the code snippet that creates a DataFrame with a MultiIndex on rows.

A

import pandas as pd
index = pd.MultiIndex.from_tuples([('a', 1), ('a', 2)])
data = {'value': [10, 20]}
df = pd.DataFrame(data, index=index)

B

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

C

import pandas as pd
data = {'A': [10, 20], 'B': [30, 40]}
df = pd.DataFrame(data)
df.set_index(['A', 'B'], inplace=True)

D

import pandas as pd
data = {'A': [10, 20], 'B': [30, 40]}
df = pd.DataFrame(data)
df.index = ['a', 'b']

💡 Hint

MultiIndex is created explicitly using pandas MultiIndex class.

Option A uses pd.MultiIndex.from_tuples to create a MultiIndex and passes it as index to the DataFrame.

Practice

1. Why is creating a DataFrame important when working with data in pandas?

easy

A. It converts data into images.

B. It automatically cleans all data errors.

C. It organizes data into rows and columns for easy analysis.

D. It deletes duplicate data without user input.

Solution

1. Step 1: Understand DataFrame structure

   A DataFrame organizes data into rows and columns, similar to a spreadsheet.

2. Step 2: Recognize the benefit for analysis

   This structure makes it easier to analyze and manipulate data clearly and simply.

3. Final Answer:

   It organizes data into rows and columns for easy analysis. -> Option C

4. Quick Check:

   DataFrames = organized data [OK]

Hint: DataFrames = tables that make data easy to handle [OK]

Common Mistakes:

• Thinking DataFrames fix data errors automatically
• Confusing DataFrames with image data
• Assuming DataFrames delete duplicates by default

2. Which of the following is the correct way to create a pandas DataFrame from a dictionary?

easy

A. pd.DataFrame({'A': [1, 2], 'B': [3, 4]})

B. pd.DataFrame['A': [1, 2], 'B': [3, 4]]

C. pd.DataFrame({'A' = [1, 2], 'B' = [3, 4]})

D. pd.DataFrame({'A': 1, 2, 'B': 3, 4})

Solution

1. Step 1: Check dictionary syntax

   The dictionary keys must be strings and values lists, using colons ':' to assign.

2. Step 2: Confirm DataFrame constructor usage

   pd.DataFrame() takes a dictionary with keys as column names and lists as column data.

3. Final Answer:

   pd.DataFrame({'A': [1, 2], 'B': [3, 4]}) -> Option A

4. Quick Check:

   Correct dict + pd.DataFrame() = pd.DataFrame({'A': [1, 2], 'B': [3, 4]}) [OK]

Hint: Use curly braces and colons inside pd.DataFrame() [OK]

Common Mistakes:

• Using square brackets instead of parentheses for DataFrame
• Using '=' instead of ':' in dictionary
• Passing single values instead of lists for column data

3. What will be the output of this code?

import pandas as pd
data = {'Name': ['Anna', 'Bob'], 'Age': [25, 30]}
df = pd.DataFrame(data)
print(df.shape)

medium

A. (2, 3)

B. (2, 2)

C. (3, 2)

D. (1, 2)

Solution

1. Step 1: Understand DataFrame shape

   DataFrame shape returns (rows, columns).

2. Step 2: Count rows and columns in data

   Data has 2 names and 2 ages, so 2 rows and 2 columns.

3. Final Answer:

   (2, 2) -> Option B

4. Quick Check:

   Rows=2, Columns=2 [OK]

Hint: Shape = (rows, columns) count [OK]

Common Mistakes:

• Confusing rows with columns count
• Assuming shape returns columns first
• Counting keys as rows

4. Identify the error in this DataFrame creation code:

import pandas as pd
data = {'X': [1, 2], 'Y': 3}
df = pd.DataFrame(data)

medium

A. Data values must be lists or arrays, not single integers.

B. Dictionary keys must be integers, not strings.

C. pd.DataFrame() cannot take a dictionary as input.

D. Missing import statement for pandas.

Solution

1. Step 1: Check data dictionary values

   Values are single integers, but DataFrame expects list-like objects for columns.

2. Step 2: Understand DataFrame input requirements

   Each column must have multiple values in a list or array to form rows.

3. Final Answer:

   Data values must be lists or arrays, not single integers. -> Option A

4. Quick Check:

   Values need to be lists [OK]

Hint: Use lists for column data, not single numbers [OK]

Common Mistakes:

• Using single values instead of lists
• Thinking keys must be integers
• Ignoring import pandas statement

5. You have nested data: {'Team': ['A', 'B'], 'Scores': [[10, 20], [15, 25]]}. How do you create a DataFrame that shows each score as a separate row with the team name repeated?

hard

A. Convert nested lists to strings before creating the DataFrame.

B. Create DataFrame directly with nested lists; pandas will flatten automatically.

C. Use pd.DataFrame({'Team': ['A', 'B'], 'Scores': [10, 20, 15, 25]}).

D. Use pd.DataFrame.explode() on the 'Scores' column after creating the DataFrame.

Solution

1. Step 1: Create DataFrame with nested lists

   Creating DataFrame with nested lists keeps 'Scores' as lists in cells.

2. Step 2: Use explode() to flatten lists into rows

   explode() expands each list element into its own row, repeating 'Team' values.

3. Final Answer:

   Use pd.DataFrame.explode() on the 'Scores' column after creating the DataFrame. -> Option D

4. Quick Check:

   explode() flattens nested lists [OK]

Hint: Use explode() to turn list elements into rows [OK]

Common Mistakes:

• Expecting pandas to flatten nested lists automatically
• Manually flattening data without explode()
• Converting lists to strings losing data structure