Why DataFrame is the core data structure in Data Analysis Python - Challenge Your Understanding

Challenge - 5 Problems

🧠 Conceptual

intermediate

Why is a DataFrame considered the core data structure in data analysis?

Choose the best reason why DataFrame is central to data analysis tasks.

A It allows storing data in rows and columns, making it easy to manipulate and analyze structured data.

B It is a tool used only for visualizing data, not for storing or manipulating it.

C It is a simple list that stores only numbers for fast calculations.

D It stores data as plain text files without any structure.

💡 Hint

Think about how you work with tables in real life, like spreadsheets.

DataFrames organize data in rows and columns, similar to spreadsheets, which makes data easy to access, filter, and analyze.

❓ Predict Output

intermediate

What is the output shape of this DataFrame?

Given the code below, what is the shape (rows, columns) of the resulting DataFrame?

Data Analysis Python

import pandas as pd

data = {'Name': ['Anna', 'Bob', 'Cara'], 'Age': [28, 34, 29], 'City': ['NY', 'LA', 'SF']}
df = pd.DataFrame(data)
print(df.shape)

A (1, 3)

B (3, 2)

C (3, 3)

D (2, 3)

💡 Hint

Count the number of rows and columns in the data dictionary.

The DataFrame has 3 rows (one for each person) and 3 columns (Name, Age, City).

❓ data_output

advanced

What is the output of filtering this DataFrame?

What rows remain after filtering the DataFrame for Age > 30?

Data Analysis Python

import pandas as pd

data = {'Name': ['Anna', 'Bob', 'Cara', 'Dan'], 'Age': [28, 34, 29, 40]}
df = pd.DataFrame(data)
filtered = df[df['Age'] > 30]
print(filtered)

A

Empty DataFrame
Columns: [Name, Age]
Index: []

B

   Name  Age
1   Bob   34
3   Dan   40

C

   Name  Age
0  Anna   28
1   Bob   34
2  Cara   29
3   Dan   40

D

   Name  Age
0  Anna   28
2  Cara   29

💡 Hint

Look for rows where Age is greater than 30.

Only Bob (34) and Dan (40) have Age values greater than 30, so only their rows remain.

❓ visualization

advanced

Which plot best shows the distribution of ages in this DataFrame?

Given a DataFrame with ages, which plot type best visualizes the age distribution?

Data Analysis Python

import pandas as pd
import matplotlib.pyplot as plt

data = {'Age': [22, 25, 29, 22, 30, 25, 28, 29, 30, 22]}
df = pd.DataFrame(data)

# Assume the plot is created here

A Histogram showing frequency of each age group

B Line plot connecting ages in order

C Scatter plot with Age on x-axis and index on y-axis

D Pie chart showing percentage of each age

💡 Hint

Think about how to show how many times each age appears.

A histogram groups ages into bins and shows how many values fall into each, making it ideal for distribution.

🔧 Debug

expert

What error does this DataFrame operation raise?

What error occurs when running this code?

Data Analysis Python

import pandas as pd

data = {'Name': ['Anna', 'Bob'], 'Age': [28, 34]}
df = pd.DataFrame(data)

# Attempt to access a non-existent column
print(df['Salary'])

A ValueError

B TypeError

C IndexError

D KeyError

💡 Hint

Check what happens when you try to access a column that is not in the DataFrame.

Accessing a column that does not exist in a DataFrame raises a KeyError.

Practice

1. Why is a DataFrame considered the core data structure in pandas for data analysis?

easy

A. Because it is used to write data to files only.

B. Because it only stores data as a list of numbers without labels.

C. Because it is a type of graph used for visualization.

D. Because it organizes data in rows and columns like a table, making it easy to understand and use.

Solution

1. Step 1: Understand DataFrame structure

   A DataFrame organizes data in rows and columns, similar to a spreadsheet or table.

2. Step 2: Recognize its role in data analysis

   This tabular structure makes it easy to view, manipulate, and analyze data efficiently.

3. Final Answer:

   Because it organizes data in rows and columns like a table, making it easy to understand and use. -> Option D

4. Quick Check:

   DataFrame = tabular data structure [OK]

Hint: Think of DataFrame as a spreadsheet table for data [OK]

Common Mistakes:

• Confusing DataFrame with simple lists or arrays
• Thinking DataFrame is only for visualization
• Believing DataFrame stores data without labels

2. Which of the following is the correct way to create a simple DataFrame in pandas?

easy

A. pd.DataFrame({'Name': ['Alice', 'Bob'], 'Age': [25, 30]})

B. pd.DataFrame['Name', 'Age'] = ['Alice', 25]

C. pd.DataFrame(Name = 'Alice', Age = 25)

D. pd.DataFrame(['Alice', 25], ['Bob', 30])

Solution

1. Step 1: Check pandas DataFrame creation syntax

   The correct way is to pass a dictionary where keys are column names and values are lists of column data.

2. Step 2: Verify each option

   pd.DataFrame({'Name': ['Alice', 'Bob'], 'Age': [25, 30]}) uses a dictionary with lists, which is correct. Others have syntax errors or wrong formats.

3. Final Answer:

   pd.DataFrame({'Name': ['Alice', 'Bob'], 'Age': [25, 30]}) -> Option A

4. Quick Check:

   DataFrame creation = dictionary of lists [OK]

Hint: Use dictionary with column names as keys to create DataFrame [OK]

Common Mistakes:

• Using square brackets instead of parentheses for DataFrame
• Passing single values instead of lists for columns
• Incorrect assignment syntax

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

import pandas as pd
data = {'City': ['NY', 'LA', 'SF'], 'Population': [8, 4, 3]}
df = pd.DataFrame(data)
print(df.loc[1, 'City'])

medium

A. NY

B. LA

C. SF

D. KeyError

Solution

1. Step 1: Understand DataFrame creation

   The DataFrame has rows indexed 0, 1, 2 with 'City' values 'NY', 'LA', 'SF' respectively.

2. Step 2: Access value at row 1, column 'City'

   Using df.loc[1, 'City'] accesses the second row's 'City' value, which is 'LA'.

3. Final Answer:

   LA -> Option B

4. Quick Check:

   Row 1 City = LA [OK]

Hint: loc[row, column] accesses exact cell by label [OK]

Common Mistakes:

• Confusing loc with iloc (integer position)
• Assuming index starts at 1 instead of 0
• Expecting KeyError due to wrong indexing

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

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

medium

A. The print statement syntax is incorrect.

B. DataFrame cannot be created without specifying columns explicitly.

C. Data dictionary values must be lists or arrays, not single values.

D. Pandas library is not imported correctly.

Solution

1. Step 1: Check data dictionary format

   The dictionary has a list for 'Name' and a single value for 'Age' instead of matching lists, which is invalid for DataFrame creation.

2. Step 2: Understand DataFrame input requirements

   DataFrame expects each column to have a list or array of values, even if only one row.

3. Final Answer:

   Data dictionary values must be lists or arrays, not single values. -> Option C

4. Quick Check:

   DataFrame columns need list values [OK]

Hint: Always use lists for column data when creating DataFrame [OK]

Common Mistakes:

• Passing single values instead of lists in data dictionary
• Assuming DataFrame can infer single values as rows
• Ignoring error messages about data format

5. You have a list of dictionaries representing sales data:

sales = [
  {'Product': 'A', 'Price': 10, 'Quantity': 2},
  {'Product': 'B', 'Price': 15, 'Quantity': 1},
  {'Product': 'A', 'Price': 10, 'Quantity': 3}
]

How can you create a DataFrame and then calculate the total sales amount for each product?

hard

A. Create DataFrame with pd.DataFrame(sales), then use df['Total'] = df['Price'] * df['Quantity'] and df.groupby('Product')['Total'].sum()

B. Create DataFrame with pd.DataFrame(sales), then use df.groupby('Product')['Price'].sum()

C. Create DataFrame with pd.DataFrame(sales), then use df.sum(axis=1) to get totals

D. Create DataFrame with pd.DataFrame(sales), then use df['Price'] * df['Quantity'] without grouping

Solution

1. Step 1: Create DataFrame from list of dictionaries

   Use pd.DataFrame(sales) to convert the list into a DataFrame with columns 'Product', 'Price', and 'Quantity'.

2. Step 2: Calculate total sales per row and group by product

   Create a new column 'Total' by multiplying 'Price' and 'Quantity'. Then group by 'Product' and sum the 'Total' values to get total sales per product.

3. Final Answer:

   Create DataFrame with pd.DataFrame(sales), then use df['Total'] = df['Price'] * df['Quantity'] and df.groupby('Product')['Total'].sum() -> Option A

4. Quick Check:

   Total sales = Price * Quantity grouped by Product [OK]

Hint: Multiply price and quantity first, then group by product [OK]

Common Mistakes:

• Summing price without considering quantity
• Not creating a total sales column before grouping
• Using sum(axis=1) which sums columns, not grouped totals