Adding and removing columns in Data Analysis Python - Interactive Code Practice

Practice - 5 Tasks

1fill in blank

easy

Complete the code to add a new column 'C' with all values set to 10.

Data Analysis Python

import pandas as pd

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

Drag options to blanks, or click blank then click option'

A'10'

B[10, 10]

Crange(2)

D10

Attempts:

3 left

💡 Hint

Common Mistakes

Assigning a single integer without a list causes the column to fill with that integer for all rows.

Assigning a string '10' creates a column of strings, not integers.

To add a new column with the same value for each row, you can assign a list with the values. Here, [10, 10] matches the number of rows.

2fill in blank

medium

Complete the code to remove the column 'B' from the DataFrame.

Data Analysis Python

import pandas as pd

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

Drag options to blanks, or click blank then click option'

Adelete

Bremove

Cdrop

Dpop

Attempts:

3 left

💡 Hint

Common Mistakes

Using remove or delete which are not pandas DataFrame methods.

Forgetting to set axis=1 to specify column removal.

The drop method removes columns or rows. To remove a column, specify the column name and axis=1.

3fill in blank

hard

Fix the error in the code to add a new column 'D' as the sum of columns 'A' and 'B'.

Data Analysis Python

import pandas as pd

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

Drag options to blanks, or click blank then click option'

A/

B-

C*

D+

Attempts:

3 left

💡 Hint

Common Mistakes

Using subtraction or multiplication instead of addition.

Using division which would not sum the columns.

To add two columns element-wise, use the plus operator +.

4fill in blank

hard

Fill both blanks to create a new column 'E' with values from column 'A' multiplied by 2, but only for rows where 'B' is greater than 3.

Data Analysis Python

import pandas as pd

df = pd.DataFrame({'A': [1, 2, 3], 'B': [3, 4, 5]})
df['E'] = df['A'] [1] 2
filtered = df[df['B'] [2] 3]
print(filtered)

Drag options to blanks, or click blank then click option'

A*

B>

C<

D+

Attempts:

3 left

💡 Hint

Common Mistakes

Using addition instead of multiplication.

Using less than operator instead of greater than for filtering.

Multiply column 'A' by 2 using *. Filter rows where 'B' is greater than 3 using >.

5fill in blank

hard

Fill all three blanks to create a dictionary with keys as uppercase column names and values as the mean of each column, but only for columns where the mean is greater than 2.

Data Analysis Python

import pandas as pd

df = pd.DataFrame({'a': [1, 3, 5], 'b': [2, 4, 6], 'c': [0, 1, 1]})
result = { [1]: df[[2]].mean() for [3] in df.columns if df[[3]].mean() > 2 }
print(result)

Drag options to blanks, or click blank then click option'

Acol.upper()

Bcol

Dcol.lower()

Attempts:

3 left

💡 Hint

Common Mistakes

Using lowercase for keys instead of uppercase.

Using wrong variable names in the comprehension.

The dictionary comprehension uses col.upper() as keys, col to select columns, and iterates over df.columns.

Practice

1. What is the correct way to add a new column named 'age' with all values set to 30 in a pandas DataFrame df?

easy

A. df.append('age', 30)

B. df.add_column('age', 30)

C. df['age'] = 30

D. df.insert_column('age', 30)

Solution

1. Step 1: Understand how to add columns in pandas

   In pandas, you add a new column by assigning a value or list to df['column_name'].

2. Step 2: Apply the correct syntax to add a column named 'age'

   Using df['age'] = 30 adds a new column 'age' with all values set to 30.

3. Final Answer:

   df['age'] = 30 -> Option C

4. Quick Check:

   Adding column = df['new_col'] = value [OK]

Hint: Use df['col_name'] = value to add columns fast [OK]

Common Mistakes:

• Using non-existent methods like add_column or insert_column
• Trying to use append which is for rows, not columns
• Forgetting to use square brackets for column name

2. Which of the following is the correct syntax to remove a column named 'salary' from a pandas DataFrame df?

easy

A. df.drop('salary', axis=1)

B. df.remove('salary')

C. df.delete_column('salary')

D. df.drop_column('salary')

Solution

1. Step 1: Recall pandas method to remove columns

   The pandas method to remove columns is drop() with axis=1 to specify columns.

2. Step 2: Use correct syntax to drop 'salary' column

   Using df.drop('salary', axis=1) removes the 'salary' column from the DataFrame.

3. Final Answer:

   df.drop('salary', axis=1) -> Option A

4. Quick Check:

   Remove column = df.drop('col', axis=1) [OK]

Hint: Use df.drop('col', axis=1) to remove columns [OK]

Common Mistakes:

• Using methods like remove or delete_column which don't exist
• Forgetting axis=1 to specify column removal
• Using drop without parentheses or wrong parameters

3. Given the DataFrame df below:

   name  score
0  Amy     90
1  Bob     80
2  Cara    85

What will be the output of this code?

df['passed'] = df['score'] >= 85
print(df)

medium

A. name score passed
0 Amy 90 False
1 Bob 80 True
2 Cara 85 False

B. name score passed
0 Amy 90 True
1 Bob 80 False
2 Cara 85 True

C. name score
0 Amy 90
1 Bob 80
2 Cara 85

D. name score passed
0 Amy 90 90
1 Bob 80 80
2 Cara 85 85

Solution

1. Step 1: Understand the condition applied to 'score'

   The expression df['score'] >= 85 creates a boolean series: True where score is 85 or more, False otherwise.

2. Step 2: Assign boolean series to new column 'passed'

   Adding df['passed'] = ... adds this boolean series as a new column.

3. Final Answer:

   DataFrame with 'passed' column showing True for scores >= 85 -> Option B

4. Quick Check:

   Boolean condition adds True/False column [OK]

Hint: Compare column with condition to add True/False column [OK]

Common Mistakes:

• Expecting numeric values instead of booleans in new column
• Not using >= operator correctly
• Printing df before adding the new column

4. What is wrong with this code snippet to remove the column 'height' from DataFrame df?

df.drop('height')

medium

A. Missing axis=1 to specify column removal

B. The column name should be in a list like ['height']

C. The method drop does not exist in pandas

D. You must assign the result back to df like df = df.drop('height')

Solution

1. Step 1: Check the drop method parameters

   By default, drop() removes rows. To remove columns, you must specify axis=1.

2. Step 2: Identify the missing axis parameter

   The code misses axis=1, so it tries to remove a row named 'height', causing an error if no such row exists.

3. Final Answer:

   Missing axis=1 to specify column removal -> Option A

4. Quick Check:

   drop columns = drop('col', axis=1) [OK]

Hint: Always add axis=1 to drop columns, not rows [OK]

Common Mistakes:

• Forgetting axis=1 causes wrong row removal or error
• Thinking drop modifies df inplace without assignment
• Using wrong data type for column name

5. You have a DataFrame df with columns ['A', 'B', 'C']. You want to create a new column 'D' which is the sum of 'A' and 'B', then remove column 'C'. Which code correctly does this in one step?

hard

A. df['D'] = df['A'] + df['B']
df.remove('C')

B. df['D'] = df['A'] + df['B']
df.drop('C', axis=1, inplace=True)

C. df['D'] = df['A'] + df['B']
df = df.drop('C', axis=1)

D. df = df.assign(D=df['A'] + df['B']).drop('C', axis=1)

Solution

1. Step 1: Create new column 'D' as sum of 'A' and 'B'

   Using assign() allows adding a new column in a chainable way without modifying df inplace.

2. Step 2: Remove column 'C' using drop() chained after assign()

   Chaining drop('C', axis=1) removes column 'C' and returns the modified DataFrame.

3. Step 3: Assign the result back to df

   Assigning the chained operations back to df updates the DataFrame correctly in one step.

4. Final Answer:

   df = df.assign(D=df['A'] + df['B']).drop('C', axis=1) -> Option D

5. Quick Check:

   Use assign + drop chained and assign back [OK]

Hint: Chain assign and drop, then assign back to df [OK]

Common Mistakes:

• Using drop without axis=1 or inplace causing no change
• Using non-existent remove method
• Not assigning chained result back to df