Sentry Answers>Python>

Select rows from a Python Pandas DataFrame based on column values

Select rows from a Python Pandas DataFrame based on column values

David Y.

The ProblemJump To Solution

How can I select rows from a DataFrame in Python Pandas based on column values? In other words, what is the DataFrame equivalent of a SELECT WHERE statement in SQL?

The Solution

This can be achieved using the DataFrame’s loc property.

To select rows with a specified column value:

Click to Copy
my_value = 3 results = my_dataframe.loc[my_dataframe["column_name"] == my_value]

To select rows that do not match a specified column value:

Click to Copy
my_value = 3 results = my_dataframe.loc[my_dataframe["column_name"] != my_value]

To select rows with a column value that matches one of a list of values:

Click to Copy
my_list = [1, 2, 3] results = my_dataframe.loc[my_dataframe["column_name"].isin(my_list)]

To select rows with a column value that falls in a range:

Click to Copy
lower_limit = 1 upper_limit = 3 my_dataframe.loc[(my_dataframe["column_name"] >= lower_limit) & (my_dataframe["column_name"] <= upper_limit)]
  • 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. Web development tips and tricks hosted by Wes Bos and Scott Tolinski

    Listen to Syntax

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.