Select rows from a Python Pandas DataFrame based on column values
David Y.
The Problem
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:
my_value = 3 results = my_dataframe.loc[my_dataframe["column_name"] == my_value]
To select rows that do not match a specified column value:
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:
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:
lower_limit = 1 upper_limit = 3 my_dataframe.loc[(my_dataframe["column_name"] >= lower_limit) & (my_dataframe["column_name"] <= upper_limit)]
Ask a questionImprove this Answer
Related Answers
- Access the index of a for loop
- Accessing the Index in a `for` Loop in Python
- Append vs. extend in Python
- Change the size of figures drawn with Matplotlib in Python
- Check if a dictionary contains a key in Python
- Check if a list is empty in Python
- Concatenate two lists in Python
- Convert a list to a string in Python
- Convert an integer to a string in Python
- Convert bytes to a string in Python
- Convert a string containing a date into datetime in Python
- Delete a file or folder in Python
- Determine the type of a Python variable
- Difference between `@staticmethod` and `@classmethod` function decorators in Python
- Extract a substring from a string in Python
- How to manage feature flags without making performance worse
- Find the current directory and executed file's directory in Python
- Get the length of a list in Python
- How do I parse a string to a float or int?
- What does `if __name__ == "__main__":` do?
- Iterate over rows in a Pandas DataFrame in Python
- Limit floats to two decimal points in Python
- Is There a List of pytz Timezones?
- Lowercase a string in Python
- Making a list out of a list of lists in Python?
- Pandas DataFrame row count in Python
- Add new keys to a dictionary in Python
- Sort a dictionary by value in Python
- Iterate over a dictionary in Python
- List files in a directory using Python
- How Do I Merge Two Dictionaries in a Single Expression (Take Union of Dictionaries)?
- How Can I Safely Create a Nested Directory?
- Python slice notation
- Does Python Have a Ternary Conditional Operator?
- What Does the `yield` Keyword in Python Do?
- Read a file line by line into a list Python
- Remove element from a list by index in Python
- Rename columns in a Python Pandas DataFrame
- Running External Programs in Python
- Select multiple columns in Python Pandas
- TypeError 'str' Object Does Not Support Item Assignment
- Time delay in Python
- UnboundLocalError Local Variable 'index' Referenced Before Assignment
- Upgrade specific `pip` packages in Python
- Use global variables in Python functions
- What are Python Metaclasses?