Sentry Answers>Python>

Delete a column from a DataFrame in Python Pandas

Delete a column from a DataFrame in Python Pandas

David Y.

The ProblemJump To Solution

How do I delete a column from a Pandas DataFrame?

The Solution

We can do this using the DataFrame.drop method:

Click to Copy
import pandas # DataFrame with three columns products = pandas.DataFrame([["apple", 1, 2], ["orange", 3, 4], ["pear", 5, 6]], columns=["product", "cost_price", "sale_price"]) print(products) # Remove the sale_price column products.drop('sale_price', axis=1, inplace=True) print(products)

This code will print the products DataFrame with three columns and then with two columns. We’ve used the following arguments in our drop method call:

  1. 'sale_price' is the name of the column to remove. We could also provide a column index (e.g. 2) or a list of indices or names to delete multiple columns.
  2. axis=1 indicates that we should remove a column (axis=0 will attempt to remove a row).
  3. inplace=True tells the method to modify the current products DataFrame rather than returning a new one with the column removed. Without this argument, we would have to reassign products as below:
Click to Copy
products = products.drop('sale_price', axis=1)
  • Sentry BlogPython Performance Testing: A Comprehensive Guide
  • Sentry BlogLogging in Python: A Developer’s Guide
  • Syntax.fm logo
    Listen to the Syntax Podcast

    Tasty treats for web developers brought to you by Sentry. Get tips and tricks from Wes Bos and Scott Tolinski.

    SEE EPISODES

Loved by over 4 million developers and more than 90,000 organizations worldwide, Sentry provides code-level observability to many of the world’s best-known companies like Disney, Peloton, Cloudflare, Eventbrite, Slack, Supercell, and Rockstar Games. Each month we process billions of exceptions from the most popular products on the internet.

© 2024 • Sentry is a registered Trademark
of Functional Software, Inc.