Altcademy - a Forbes magazine logo Best Coding Bootcamp 2023

How to select columns in Pandas

Understanding Pandas DataFrames

Before we dive into the specifics of selecting columns, it's essential to grasp what a DataFrame is. In Pandas, a DataFrame is a two-dimensional, size-mutable, and potentially heterogeneous tabular data structure with labeled axes (rows and columns). Think of it as a spreadsheet or a table in a relational database.

Selecting a Single Column

To select a single column from a DataFrame, you can use a simple bracket [] notation with the name of the column as a string. This is similar to how you might pick a card from a deck by naming the card you want.

import pandas as pd

# Sample DataFrame
data = {'Name': ['Alice', 'Bob', 'Charlie'], 'Age': [25, 30, 35], 'City': ['New York', 'Paris', 'London']}
df = pd.DataFrame(data)

# Selecting the 'Name' column
names = df['Name']

This will output the 'Name' column:

0      Alice
1        Bob
2    Charlie
Name: Name, dtype: object

Selecting Multiple Columns

If you want to select more than one column, you can pass a list of column names to the brackets. This is like picking multiple items from a menu by pointing out each one you're interested in.

# Selecting 'Name' and 'Age' columns
subset = df[['Name', 'Age']]

The output will be:

      Name  Age
0    Alice   25
1      Bob   30
2  Charlie   35

Using .loc[] to Select Columns

The .loc[] method is a powerful tool in Pandas that allows you to select data based on label information. To select columns using .loc[], you can do the following:

# Selecting 'Name' and 'City' columns using .loc
subset_loc = df.loc[:, ['Name', 'City']]

The colon : before the comma indicates that you want to select all rows, and the list after the comma specifies the columns you want.

Using .iloc[] for Position-Based Selection

When you want to select columns based on their integer location, rather than their label, you can use .iloc[]. This is like telling a friend which item you want by counting from the start of the menu.

# Selecting the first and third columns using .iloc
subset_iloc = df.iloc[:, [0, 2]]

This will output the first and third columns:

      Name     City
0    Alice  New York
1      Bob    Paris
2  Charlie   London

Using .loc[] and .iloc[] with Slicing

Both .loc[] and .iloc[] support slicing. Slicing is a way to select a range of items, like choosing a continuous group of pages in a book.

# Using slicing with .loc to select columns 'Name' through 'Age'
subset_loc_slice = df.loc[:, 'Name':'Age']

And with .iloc[], you can use slicing by specifying the range of column indices:

# Using slicing with .iloc to select the first two columns
subset_iloc_slice = df.iloc[:, 0:2]

Conditional Selection

Sometimes, you may want to select columns based on certain conditions. For example, you might only be interested in numeric data. This is akin to choosing dishes from a menu based on dietary preferences.

# Selecting columns with numeric data
numeric_data = df.select_dtypes(include=[int, float])

In this case, select_dtypes is a method that allows you to filter the DataFrame's columns based on their data type.

Using Methods to Select Columns

Pandas also provides methods like .filter() which can be used to select columns based on specific criteria, such as regular expressions or like patterns. This is like using a search function to quickly find all items on a menu that contain a particular ingredient.

# Using .filter() to select columns containing 'Name' in their column name
filtered_columns = df.filter(like='Name')

Tips and Tricks for Column Selection

When working with DataFrames, you may encounter situations where you need to select columns dynamically. For instance, you might want to select all columns except one. Here's a trick to do that:

# Selecting all columns except 'Age'
columns_except_age = df.loc[:, df.columns != 'Age']

Understanding Common Errors

As you learn to select columns, you'll likely run into some common errors. For example, if you misspell a column name or try to select a column that doesn't exist, you'll get a KeyError. It's like asking for a dish that's not on the menu—the restaurant can't serve what it doesn't have.

# Attempting to select a non-existent column
    non_existent = df['Height']
except KeyError as e:
    print(f"Error: {e}")

This will output an error message indicating that the 'Height' column does not exist.

Intuitions and Analogies

Selecting columns in Pandas can be thought of as the process of choosing ingredients for a recipe. Just as you'd pick specific ingredients based on what you're planning to cook, you select specific columns based on the data analysis you intend to perform.


Exploring a DataFrame and selecting the columns you need is like setting the stage before a performance. It's about arranging your tools and ingredients before you start the actual cooking. This preparatory step is crucial because it allows you to focus on the data that matters most for your analysis.

As you become more familiar with these techniques, you'll find that selecting columns in Pandas becomes second nature. It's a foundational skill that will serve you well on your programming journey, much like learning to read a recipe before you start cooking. With practice, you'll be able to swiftly and confidently prepare your data for any analysis or visualization task that comes your way.