How to filter a DataFrame by column value in one line

Learn how to filter a pandas DataFrame by column value in one line. A beginner-friendly guide covering setup, data loading, and clean filtering techniques


Filtering a DataFrame by a column value in one line is one of the most practical operations in Python data workflows, especially when using pandas.


You start by importing pandas and loading your dataset into a DataFrame—typically from a CSV or database. Once loaded, it’s good practice to quickly inspect the structure using .head() or .info() so you understand the column names and data types. 



Here is the world bank dataset: https://data.worldbank.org/indicator/NY.GDP.MKTP.CD?locations=A9


This step is critical because filtering depends on exact column references and compatible data types.


Next, identify the column you want to filter and the specific value you’re targeting. For example, suppose you want only rows where the country column equals "Kenya".


The actual filtering happens using boolean indexing, where pandas evaluates a condition and returns only the rows that meet it. This can be done in a single, clean line:

filtered_df = df[df['Country Name'] == 'Kenya']

Here’s what’s happening under the hood:

  • df['Country Name'] == 'Kenya' creates a boolean series (True/False for each row)

  • df[...] uses that series to keep only the matching rows

That’s it—one line, fully vectorized, and highly efficient. 

This pattern scales across datasets and becomes the foundation for more advanced filtering like multiple conditions, ranges, and text matching.



Advance Your Career With 16 Python Projects in Data & ML — All for $288.



Comments

Post a Comment

Popular posts from this blog

How to Filter Rows Using Boolean Indexing in Pandas (Afrobarometer Kenya Dataset)

Image
Learn how to filter rows in Pandas using boolean indexing with real Afrobarometer Kenya survey data. Build clean, ML-ready features with practical examples. Why Boolean Indexing Matters in Feature Engineering In Pandas, boolean indexing is how you select the exact slices of data that become features .  In Module 04 of the course, you’ll no longer exploring—you’ll be deciding what the model sees. In this case, with a dataset like Afrobarometer Kenya Round 10 , filtering is how you: Remove invalid survey responses Isolate specific demographic groups Prepare clean subsets for encoding Load and Inspect the Dataset This assumes that you’ve already converted the .sav file and loaded it: import pandas as pd df = pd.read_csv("afrobarometer_kenya_r10.csv") df.head() 1. Filter Valid Survey Responses Survey datasets contain codes like -1 , 8 , 9 for missing or “Don’t know”. df = df[df["trust_president"] >= 0] This keeps only valid numeric responses for trust in the pres...
Read more

How to Decide Whether to Drop or Fill Missing Value

Image
Learn how to decide whether to drop or fill missing values in datasets using practical, production-ready strategies. Understand trade-offs, bias risks, and data engineering best practices. Missing values are not just a data cleaning nuisance—they’re a signal . The decision to drop or fill them has direct consequences on model performance, statistical validity, and downstream business decisions. If you treat missing data casually, you risk introducing bias, noise, or false confidence into your analysis. This guide gives you a structured, decision-oriented framework you can use in real-world data workflows. 1. First Principle: Understand Why Data Is Missing Before deciding what to do , you need to understand why it’s missing . In data science, this is formalized into three categories: 1. Missing Completely at Random (MCAR) Missingness has no relationship to any variable. Example: A sensor randomly fails. Implication: Safe to drop rows without introducing bias. 2. Missing at Random (MAR...
Read more

How to create your first line chart with World Bank Kenya GDP data

Image
Learn how to pull Kenya GDP data directly from the World Bank API and create your first line chart in Python using pandas and matplotlib—clean, fast, and reproducible. 1. Install Required Libraries !pip install pandas matplotlib wbdata 2. Import Dependencies import pandas as pd import matplotlib.pyplot as plt import wbdata import datetime 3. Define the Data You Need World Bank uses indicators . For GDP (current US$), the indicator is: NY.GDP.MKTP.CD Set the country to Kenya ( KEN ) and define a time range: indicator = {'NY.GDP.MKTP.CD': 'gdp'} data_date = (datetime.datetime(2000, 1, 1), datetime.datetime(2023, 1, 1)) 4. Fetch Data from World Bank df = wbdata.get_dataframe(indicator, country='KEN', data_date=data_date) 5. Clean and Prepare Data df = df.reset_index() df['date'] = pd.to_datetime(df['date']) df = df.sort_values('date') print (df) Explanation df = df.reset_index() → Converts the index into a normal column and resets ro...
Read more