How to only get the column values that are in a list with Pandas using Python
Land Your First Data Science Job
A proven roadmap to prepare for $75K+ entry-level data roles. Perfect for Data Scientist ready to level up their career.
Here is a quick and efficient way to filter the column values that are contained in a list using the DataFrame.isin() method.
Imagine that we have a DataFrame containing GDP observations per country for 2021.
But now we only want to select a few countries by name.
E.g. filter for Germany and France
Here is the code
# To work with dataframes
import pandas as pd
# We create our sample dataframe
df = pd.DataFrame({"Country" : ["Germany", "USA", "France"],
"Temperatures" : [3332000000,
20839000000,
2603000000]})
# A list of values we are interested in
selected_countries = ["Germany","France"]
# We filter only for the countries that are in the provided list
mask = df["Country"].isin(selected_countries)
# We print the dataframe
print(df[mask])
Here you are! You know now how to only get the column values that are in a list with Pandas using Python.
More on DataFrames
If you want to know more about DataFrame and Pandas. Check out the other articles I wrote on the topic, just here :
Land Your First Data Science Job
A proven roadmap to prepare for $75K+ entry-level data roles. Perfect for Data Scientist ready to level up their career.
Related Articles
Continue your learning journey with these related topics
Master Data Science in Days, Not Months 🚀
Skip the theoretical rabbit holes. Get practical data science skills delivered in bite-sized lessons – Approach used by real data scientist. Not bookworms. 📚