Selecting multiple columns in Pandas - Cheat Sheet & Quick Revision

Recall & Review

beginner

How do you select multiple columns from a pandas DataFrame?

You select multiple columns by passing a list of column names inside double square brackets, like df[["col1", "col2"]].

beginner

What type of object is returned when selecting multiple columns from a DataFrame?

Selecting multiple columns returns a new DataFrame containing only those columns.

intermediate

What happens if you select a single column using double square brackets, like df[["col1"]]?

You get a DataFrame with one column, not a Series. Single square brackets return a Series, double brackets return a DataFrame.

intermediate

Can you select columns by position using the same method as selecting by name?

No, selecting by position uses df.iloc, while selecting by name uses df[[...]].

beginner

Why might you want to select multiple columns from a DataFrame?

To focus analysis on specific data, reduce memory use, or prepare data for visualization or modeling.

Which syntax correctly selects columns 'A' and 'B' from DataFrame df?

A df[['A', 'B']]

B df['A', 'B']

C df.loc['A', 'B']

D df.iloc['A', 'B']

Use double square brackets with a list of column names to select multiple columns.

What type of object is returned by df[['A', 'B']]?

A DataFrame

B Series

C List

D Dictionary

Selecting multiple columns returns a DataFrame.

How do you select columns by position instead of name?

A df[['0', '1']]

B df.iloc[:, [0, 1]]

C df.select([0, 1])

D df.loc[0, 1]

Use df.iloc with column positions inside a list to select columns by position.

What does df[['A']] return?

A An error

B A Series

C A list

D A DataFrame with one column

Double brackets return a DataFrame, even if selecting one column.

Why is selecting multiple columns useful?

A To change data types

B To delete data

C To focus on specific data for analysis

D To rename columns

Selecting columns helps focus on relevant data for analysis or visualization.

Explain how to select multiple columns from a pandas DataFrame and what the result looks like.

Think about how you pick multiple items from a list.

Describe the difference between selecting a single column with single brackets and double brackets in pandas.

Consider the type of object returned.

Practice

1. What is the correct way to select multiple columns 'A' and 'B' from a pandas DataFrame df?

easy

A. df.loc['A', 'B']

B. df['A', 'B']

C. df[['A', 'B']]

D. df['A' & 'B']

Solution

1. Step 1: Understand pandas column selection syntax

   To select multiple columns, pandas requires a list of column names inside double brackets.

2. Step 2: Apply correct syntax for multiple columns

   Using df[['A', 'B']] correctly selects columns 'A' and 'B'. Other options are invalid syntax or select rows instead.

3. Final Answer:

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

4. Quick Check:

   Multiple columns = list inside double brackets [OK]

Hint: Use double brackets with a list for multiple columns [OK]

Common Mistakes:

• Using single brackets with comma-separated names
• Using bitwise operators like & between column names
• Using .loc incorrectly for column selection

2. Which of the following is the correct syntax to select columns 'X' and 'Y' from DataFrame df?

easy

A. df.loc[:, 'X', 'Y']

B. df.loc['X', 'Y']

C. df['X', 'Y']

D. df[['X', 'Y']]

Solution

1. Step 1: Recall syntax for selecting multiple columns

   Using double brackets with a list of column names is the standard way: df[['X', 'Y']].

2. Step 2: Check other options for correctness

   df.loc[:, 'X', 'Y'] causes TypeError (invalid arguments to loc); df['X', 'Y'] raises KeyError (tuple not a column); df.loc['X', 'Y'] is incorrect syntax.

3. Final Answer:

   df[['X', 'Y']] -> Option D

4. Quick Check:

   Double brackets with list = correct syntax [OK]

Hint: Remember: double brackets with list select multiple columns [OK]

Common Mistakes:

• Using single brackets with multiple column names
• Confusing .loc row and column selection syntax
• Using tuple instead of list inside brackets

3. Given the DataFrame df below:

   A  B  C
0  1  4  7
1  2  5  8
2  3  6  9

What is the output of df[['B', 'C']]?

medium

A. B C
0 4 7
1 5 8
2 6 9

B. A B
0 1 4
1 2 5
2 3 6

C. A C
0 1 7
1 2 8
2 3 9

D. KeyError

Solution

1. Step 1: Identify columns requested

   The code selects columns 'B' and 'C' from the DataFrame.

2. Step 2: Extract columns 'B' and 'C' values

   Columns 'B' and 'C' have values [[4,7],[5,8],[6,9]] for rows 0,1,2 respectively.

3. Final Answer:

   B C 0 4 7 1 5 8 2 6 9 -> Option A

4. Quick Check:

   Selected columns 'B' and 'C' output matches

      B  C
   0  4  7
   1  5  8
   2  6  9

   [OK]

Hint: Selected columns show only those columns with all rows [OK]

Common Mistakes:

• Confusing columns selected and expecting all columns
• Getting KeyError by misspelling column names
• Selecting rows instead of columns

4. What error will occur with this code?

df = pd.DataFrame({'X':[1,2], 'Y':[3,4]})
selected = df['X', 'Y']

medium

A. TypeError

B. KeyError

C. SyntaxError

D. No error, returns both columns

Solution

1. Step 1: Analyze the indexing syntax

   The code uses single brackets with a comma, creating a tuple key ('X', 'Y').

2. Step 2: Understand pandas error type

   This raises KeyError because pandas looks for a column named ('X', 'Y'), which doesn't exist.

3. Final Answer:

   KeyError -> Option B

4. Quick Check:

   Tuple key in single brackets causes KeyError [OK]

Hint: Use list inside double brackets, not comma in single brackets [OK]

Common Mistakes:

• Using comma inside single brackets for multiple columns
• Expecting no error with wrong syntax
• Confusing TypeError with KeyError

5. You have a DataFrame df with columns ['Name', 'Age', 'City', 'Salary']. You want to create a new DataFrame df_selected with only 'Name' and 'Salary'. Which code correctly does this and also resets the index?

hard

A. df_selected = df[['Name', 'Salary']].reset_index(drop=True)

B. df_selected = df.loc[:, ['Name', 'Salary']].reset_index()

C. df_selected = df[['Name', 'Salary']].reset_index()

D. df_selected = df[['Name', 'Salary']].reset_index(drop=False)

Solution

1. Step 1: Select columns 'Name' and 'Salary'

   Use double brackets with list: df[['Name', 'Salary']] to select these columns.

2. Step 2: Reset index without adding old index as column

   Use reset_index(drop=True) to reset index and drop old index column.

3. Step 3: Check other options

   df.loc[:, ['Name', 'Salary']].reset_index() resets index but keeps old index as column. Using reset_index() without drop=True (default behavior) keeps the old index as a new column.

4. Final Answer:

   df_selected = df[['Name', 'Salary']].reset_index(drop=True) -> Option A

5. Quick Check:

   Select columns + reset index drop old index = df_selected = df[['Name', 'Salary']].reset_index(drop=True) [OK]

Hint: Select columns with double brackets, reset index with drop=True [OK]

Common Mistakes:

• Forgetting to use double brackets for multiple columns
• Not using drop=True in reset_index to avoid old index column
• Using .loc without understanding index reset effect