Arithmetic operations on columns in Pandas - Cheat Sheet & Quick Revision

Recall & Review

beginner

What does adding two columns in a pandas DataFrame do?

It adds the values in each row from the two columns and creates a new column with these sums.

beginner

How do you subtract one column from another in pandas?

You use the minus (-) operator between the two columns, like df['col1'] - df['col2'].

beginner

What happens if you multiply a column by a number in pandas?

Each value in the column is multiplied by that number, scaling the entire column.

beginner

How can you create a new column that is the result of dividing one column by another?

Use the division operator (/) between the two columns, like df['col1'] / df['col2'].

beginner

Why is it useful to perform arithmetic operations on DataFrame columns?

It helps to create new insights or features by combining or transforming existing data easily.

What does df['A'] + df['B'] do in pandas?

A Creates a list of columns A and B

B Concatenates columns A and B as strings

C Deletes columns A and B

D Adds values of columns A and B row-wise

Adding two columns sums their values for each row.

How do you multiply all values in column 'X' by 10?

A df['X'] - 10

B df['X'] + 10

C df['X'] * 10

D df['X'] / 10

Multiplying a column by 10 scales each value by 10.

Which operator divides one column by another in pandas?

A +

B /

C *

D -

The division operator (/) divides values of one column by another.

What will df['C'] = df['A'] - df['B'] do?

A Create column C with row-wise difference of A and B

B Add columns A and B into C

C Multiply columns A and B into C

D Delete columns A and B

Subtracting columns creates a new column with their difference.

Why perform arithmetic on DataFrame columns?

A To create new data insights

B To delete data

C To rename columns

D To sort data

Arithmetic operations help generate new useful data from existing columns.

Explain how to add two columns in a pandas DataFrame and why you might do this.

Think about combining data from two columns to get a total.

Describe how to multiply a DataFrame column by a number and what effect this has on the data.

Consider how changing each value by a fixed number changes the data.

Practice

1. What does the expression df['total'] = df['price'] + df['tax'] do in pandas?

easy

A. Multiplies the 'price' and 'tax' columns and stores in 'total'.

B. Deletes the 'price' and 'tax' columns from the DataFrame.

C. Filters rows where 'price' plus 'tax' is greater than zero.

D. Creates a new column 'total' by adding 'price' and 'tax' columns element-wise.

Solution

1. Step 1: Understand the operation on columns

   The expression adds values from 'price' and 'tax' columns for each row.

2. Step 2: Assign result to new column

   The sum is stored in a new column named 'total' in the DataFrame.

3. Final Answer:

   Creates a new column 'total' by adding 'price' and 'tax' columns element-wise. -> Option D

4. Quick Check:

   Adding columns creates new column with sums [OK]

Hint: Adding columns creates a new column with their sums [OK]

Common Mistakes:

• Thinking it multiplies instead of adds
• Assuming it deletes columns
• Confusing assignment with filtering

2. Which of the following is the correct syntax to subtract column 'B' from column 'A' in a pandas DataFrame df and store the result in a new column 'C'?

easy

A. df['C'] = df['A'] - df['B']

B. df['C'] = df['A'] + df['B']

C. df['C'] = df['A'] * df['B']

D. df['C'] = df['A'] / df['B']

Solution

1. Step 1: Identify the subtraction operation

   We want to subtract column 'B' from 'A', so the operator is '-'.

2. Step 2: Assign the result to new column 'C'

   Use df['C'] = df['A'] - df['B'] to store the difference.

3. Final Answer:

   df['C'] = df['A'] - df['B'] -> Option A

4. Quick Check:

   Subtraction uses '-' operator between columns [OK]

Hint: Use '-' operator between columns to subtract [OK]

Common Mistakes:

• Using '+' instead of '-'
• Using '*' or '/' operators incorrectly
• Confusing column names or assignment

3. Given the DataFrame:

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

What is the output?

medium

A. [6, 20]

B. [5, 9]

C. [2, 4]

D. [3, 5]

Solution

1. Step 1: Multiply columns 'A' and 'B' element-wise

   Row 0: 2 * 3 = 6, Row 1: 4 * 5 = 20.

2. Step 2: Print the new column 'C'

   The output shows the values [6, 20] for the two rows.

3. Final Answer:

   [6, 20] -> Option A

4. Quick Check:

   Element-wise multiplication = [6, 20] [OK]

Hint: Multiply columns element-wise to get new column values [OK]

Common Mistakes:

• Adding instead of multiplying
• Printing wrong column
• Confusing list output format

4. What is wrong with this code snippet?

df['D'] = df['A'] + '5'

Assuming 'A' is a numeric column.

medium

A. It will create a new column 'D' with all values as '5'.

B. It will add 5 to each value in column 'A' correctly.

C. It will raise a TypeError because you cannot add a string to a numeric column.

D. It will concatenate the string '5' to each numeric value in 'A'.

Solution

1. Step 1: Check data types involved in addition

   Column 'A' is numeric, but '5' is a string, so adding them causes a type mismatch.

2. Step 2: Understand pandas behavior on mixed types

   Pandas cannot add string to numeric directly, so it raises a TypeError.

3. Final Answer:

   It will raise a TypeError because you cannot add a string to a numeric column. -> Option C

4. Quick Check:

   Adding string to number causes TypeError [OK]

Hint: Check data types before arithmetic to avoid errors [OK]

Common Mistakes:

• Assuming string '5' is converted automatically
• Expecting concatenation with numbers
• Ignoring error messages

5. You have a DataFrame df with columns 'cost' and 'discount'. You want to create a new column 'final_price' that subtracts the discount from cost and then applies a 10% tax on the result. Which code correctly does this?

hard

A. df['final_price'] = df['cost'] * 1.10 - df['discount']

B. df['final_price'] = (df['cost'] - df['discount']) * 1.10

C. df['final_price'] = df['cost'] - (df['discount'] * 1.10)

D. df['final_price'] = df['cost'] - df['discount'] * 1.10

Solution

1. Step 1: Subtract discount from cost

   Calculate the price after discount: df['cost'] - df['discount'].

2. Step 2: Apply 10% tax on discounted price

   Multiply the result by 1.10 to add 10% tax.

3. Step 3: Assign to 'final_price'

   Store the final value in the new column 'final_price'.

4. Final Answer:

   df['final_price'] = (df['cost'] - df['discount']) * 1.10 -> Option B

5. Quick Check:

   Subtract then multiply for tax = correct formula [OK]

Hint: Use parentheses to control order: subtract first, then multiply [OK]

Common Mistakes:

• Missing parentheses causing wrong order
• Multiplying discount instead of discounted price
• Applying tax before subtracting discount