Altcademy - a Forbes magazine logo Best Coding Bootcamp 2023

How to delete columns in Pandas

Understanding DataFrames in Pandas

Before we dive into the process of deleting columns in Pandas, let's get a grasp on what a DataFrame is. Think of a DataFrame as a table, much like the kind you'd find in a spreadsheet. It has rows and columns, where each column can be thought of as a different attribute or feature, and each row is an individual record or data point.

Setting Up Your Environment

To get started, you'll need to have Pandas installed in your Python environment. If you haven't done this yet, you can install it using pip, which is a package installer for Python. Just run the following command in your terminal or command prompt:

pip install pandas

Now, let's import Pandas in our Python script or Jupyter notebook:

import pandas as pd

The pd is a common alias for Pandas. It's like giving a nickname to Pandas so we don't have to type 'pandas' every time we want to use a function from the library.

Creating a Simple DataFrame

Before we can delete columns, we need a DataFrame to work with. Let's create a simple one:

data = {
    'Name': ['Alice', 'Bob', 'Charlie'],
    'Age': [25, 30, 35],
    'City': ['New York', 'Paris', 'London']

df = pd.DataFrame(data)

This code creates a DataFrame with three columns: 'Name', 'Age', and 'City', and three rows of data.

Deleting Columns Using drop

Pandas provides a method called drop that we can use to remove columns. The drop method requires at least one parameter, which is the label or labels (names) of the columns you want to remove.

Deleting a Single Column

To delete a single column, you can pass the column name as a string to the drop method:

df = df.drop('Age', axis=1)

Here, axis=1 specifies that we want to drop a column, not a row. If you think of axis as dimensions, axis=0 represents rows (horizontal), and axis=1 represents columns (vertical).

Deleting Multiple Columns

To delete multiple columns, you can pass a list of column names:

df = df.drop(['Age', 'City'], axis=1)

This will remove both the 'Age' and 'City' columns from the DataFrame.

Using inplace Parameter

If you don't want to assign the result to a new variable, you can use the inplace=True parameter, which will modify the DataFrame in place:

df.drop('Age', axis=1, inplace=True)

After running this code, df will no longer include the 'Age' column, and you don't need to assign the result back to df.

Deleting Columns Using del

Another way to remove a column from a DataFrame is by using the del keyword. This method is more straightforward but can only be used to delete one column at a time:

del df['City']

After executing this code, the 'City' column will be removed from df.

Deleting Columns Based on Condition

Sometimes you may want to delete columns based on a specific condition. For example, you might want to remove all columns where all values are NaN (Not a Number, which is Pandas' way of indicating missing data).

Here's how you could do that:

df = df.dropna(axis=1, how='all')

The dropna method drops rows or columns with missing data. The axis=1 argument tells Pandas to look at columns, and how='all' specifies that only columns where all values are NaN should be dropped.

Renaming Columns After Deletion

After deleting columns, you might end up with a DataFrame that has column names that don't make sense anymore. You can rename columns using the rename method:

df = df.rename(columns={'Name': 'FirstName'})

This renames the 'Name' column to 'FirstName'.


Deleting columns in Pandas is like tidying up a room; it helps you focus on the things you need and removes unnecessary clutter. Whether you're pruning your dataset to make it more manageable or preparing it for analysis, knowing how to efficiently delete columns is a valuable skill in data manipulation.

Remember, the drop method is your multi-tool, capable of handling single or multiple columns and offering the flexibility of in-place operations. The del keyword is like a quick swipe that removes a column instantly. And, if you ever need to clean up after the deletion, renaming columns can help maintain clarity in your dataset.

As you continue your programming journey, you'll find that these operations become second nature. Just like organizing a bookshelf becomes easier the more you do it, manipulating data in Pandas will become more intuitive with practice. Keep experimenting, keep learning, and soon you'll be handling data with the finesse of a seasoned data scientist.