Altcademy - a Forbes magazine logo Best Coding Bootcamp 2023

How to drop columns in Pandas

Understanding DataFrames in Pandas

Imagine a DataFrame in Pandas as a big table, much like an Excel spreadsheet, where your data is neatly organized in rows and columns. Each column represents a variable or attribute of the data you're working with, and each row represents an individual record or entry.

Sometimes, you might find that not all the columns in your DataFrame are useful for your analysis. Maybe they contain redundant information, or perhaps they're just not relevant to the question you're trying to answer. In these cases, it's often a good idea to tidy up your DataFrame by removing these unnecessary columns. This is where the concept of "dropping" columns comes into play.

Dropping Columns: The Basics

To drop a column in Pandas, you use the drop method. This method allows you to specify the labels of the columns you want to remove from your DataFrame. Here's a basic example:

import pandas as pd

# Create a simple DataFrame
df = pd.DataFrame({
    'A': [1, 2, 3],
    'B': [4, 5, 6],
    'C': [7, 8, 9]

# Drop column 'B'
df = df.drop('B', axis=1)


The axis=1 parameter tells Pandas that you want to drop a column, not a row (which would be axis=0). The result of this operation would be:

   A  C
0  1  7
1  2  8
2  3  9

As you can see, column 'B' has been successfully removed from the DataFrame.

Removing Multiple Columns

What if you want to drop more than one column? No problem! You can pass a list of column names to the drop method:

# Drop columns 'A' and 'C'
df = df.drop(['A', 'C'], axis=1)


This will leave you with an empty DataFrame since we've dropped all the columns, but in a real-world scenario, you would typically have more columns to work with.

In-Place Dropping

By default, when you use the drop method, Pandas returns a new DataFrame without the dropped columns, and the original DataFrame remains unchanged. If you want to modify the original DataFrame directly, you can use the inplace=True parameter:

# Drop column 'B' in place
df.drop('B', axis=1, inplace=True)

Now, df itself is updated, and there's no need to assign the result back to df.

Handling Errors When Dropping Columns

If you try to drop a column that doesn't exist in your DataFrame, Pandas will raise a KeyError. To handle this situation gracefully, you can set the errors parameter to 'ignore':

# Attempt to drop a non-existent column
df.drop('D', axis=1, errors='ignore')

This tells Pandas to do nothing if the column isn't found, avoiding the KeyError.

Using the pop Method

Another way to remove a column from a DataFrame is by using the pop method. Unlike drop, pop both removes the column and returns its values. This can be useful if you want to work with the data from the dropped column immediately after removing it.

# Pop column 'C'
popped_column = df.pop('C')


Dropping Columns Based on Column Names

Sometimes, you might want to drop columns based on a pattern or a specific substring in their names. You can achieve this by using string methods combined with a boolean indexing:

# Drop columns that contain the letter 'A'
df = df[df.columns[~df.columns.str.contains('A')]]

Here, df.columns.str.contains('A') creates a boolean array that is True for column names containing 'A'. The ~ operator inverts the boolean array, and df.columns[...] selects the columns where the condition is False.

Dropping Columns with Null Values

In data analysis, you might encounter columns with many missing values (null values). If a column has too many nulls, it might not be useful for your analysis. Pandas makes it easy to drop such columns using the dropna method with the thresh parameter:

# Drop columns with more than 2 null values
df.dropna(axis=1, thresh=len(df) - 2)

This code snippet will drop any column that has more than two null values.

Conclusion: Tidying Up Your Data

Dropping columns is like pruning a garden: it's about removing the parts that are no longer serving a purpose to allow the rest of your garden (or data) to thrive. When you're working with large datasets, it's crucial to keep only the data that adds value to your analysis. By learning how to drop columns in Pandas effectively, you streamline your dataset and make your data analysis process more efficient and focused.

Remember, the key to becoming proficient in data manipulation is practice. Don't hesitate to experiment with these techniques using your datasets. As you become more comfortable with dropping columns, you'll find that it's just one of the many powerful tools Pandas offers to help you transform and understand your data. Happy data cleaning!