Select a Single Column
You have a DataFrame df with employee data. Select only the name column and store it in result. Available Data:...
pandasbasicsselection
Beginner
2m5 pts
17100%
Python Practice Hub
pandas DataFrame Practice Questions
Select, group, merge and derive columns on a table in Python, without loops.
51 questionsFreeRuns in your browser
- loc / iloc
- Select rows and columns by label or position.
- groupby + agg
- Group rows and summarise each group.
- merge
- Join two DataFrames, like a SQL join.
What you will practice
- Select and filter with loc, iloc and boolean masks instead of chained indexing
- Aggregate with groupby().agg() and reshape the result with reset_index and pivot_table
- Join frames using merge with the right how= and on= arguments, and check the row count afterwards
- Replace row-by-row loops and apply() with vectorised column arithmetic and np.where
- Handle missing data with isna, fillna and dropna before the numbers reach a report
Read the full overview
A pandas DataFrame is a labelled, two-dimensional table in Python, and most analysis with it comes down to four moves: select the rows and columns you need, group and aggregate them, merge in another table, and derive new columns without looping. Doing that vectorised, with whole-column operations instead of a Python for loop, is what keeps the code both short and fast. These questions run real pandas in the browser against seeded DataFrames, so the output you check is the actual frame.
51 questionsPage 1 of 4
Select Multiple Columns
Select the name and salary columns from the employee DataFrame. Available Data: DataFrame df with columns: name,...
pandasbasicsselectionBeginner2m5 pts15100%Filter Rows by Condition
Filter the employee DataFrame to find all employees in the Engineering department. Available Data: DataFrame df with...
pandasbasicsfilteringBeginner3m5 pts18100%Filter with Multiple Conditions
Find employees who are in Engineering AND earn more than $100,000. Available Data: DataFrame df with columns: name,...
pandasbasicsfilteringBeginner3m10 pts14100%Sort DataFrame by Column
Sort the employee DataFrame by salary in descending order. Available Data: DataFrame df with columns: name, department,...
pandasbasicssortingBeginner3m5 pts14100%Get Top N Rows
Get the top 3 highest-paid employees from the DataFrame. Available Data: DataFrame df with columns: name, department,...
pandasbasicssortingBeginner3m10 pts12100%Count Rows and Columns
Find the number of rows in the employee DataFrame. Available Data: DataFrame df with columns: name, department, salary,...
pandasbasicsBeginner2m5 pts12100%Find Unique Values
Find all unique departments in the employee DataFrame. Available Data: DataFrame df with columns: name, department,...
pandasbasicsBeginner3m5 pts11100%Add a New Column
Add a new column called bonus to the employee DataFrame. The bonus should be 10% of each employee's salary. Available...
pandasbasicstransformationBeginner3m10 pts11100%Count Values per Category
Count how many employees are in each department. Available Data: DataFrame df with columns: name, department, salary,...
pandasbasicsaggregationBeginner2m5 pts10100%Calculate Basic Statistics
Find the mean salary across all employees. Available Data: DataFrame df with columns: name, department, salary, age,...
pandasbasicsstatisticsBeginner2m5 pts11100%Rename Columns
Rename the columns of the employee DataFrame: name to employee_name and salary to annual_salary. Available Data:...
pandasbasicstransformationBeginner3m5 pts990%Set a Column as Index
Set the name column as the index of the DataFrame. Available Data: DataFrame df with columns: name, department, salary,...
pandasbasicsindexingBeginner2m5 pts9100%Filter with isin()
Find employees who work in either Engineering or HR. Available Data: DataFrame df with columns: name, department,...
pandasbasicsfilteringBeginner3m10 pts9100%Create a DataFrame from Scratch
Create a DataFrame with the following data: fruitquantityprice Apple101.5 Banana250.5 Cherry153.0 Requirements: Create...
pandasbasicscreationBeginner3m5 pts9100%
FAQ
pandas DataFrame practice: common questions
Short answers to what people get stuck on most. Tap a question to expand it.
How does pandas treat NaN in groupby sums and means?
Aggregations skip NaN by default, so a group still returns a total from its non-missing values.
- sum(), mean() and count() all ignore NaN.
- An all-NaN group sums to 0 unless you pass min_count=1 (shown here).
- Check df.isna().sum() first so you know how much data is being skipped.
SQL
df.groupby("city")["sales"].sum() # all-NaN group -> 0.0
df.groupby("city")["sales"].sum(min_count=1) # all-NaN group -> NaNWhat is the difference between loc and iloc?
loc selects by label, iloc selects by integer position.
- loc uses index values and column names, and slices include the end label.
- iloc uses positions, and slices exclude the end like normal Python.
- loc also takes boolean masks, the usual way to filter rows (shown here).
- Avoid chained indexing like df[mask]["col"] = x: it may assign to a copy (SettingWithCopyWarning).
SQL
df.loc[df["sales"] > 100, "city"] # label + boolean mask
df.iloc[0:2, 0] # first two rows, first columnHow do I join two DataFrames?
Use merge, which works like a SQL join. Choose how: inner, left, right or outer.
- Check the row count after merging: a many-to-many match multiplies rows.
- validate="many_to_one" makes pandas raise an error if the right key is not unique (shown here).
- Use pd.concat instead to stack DataFrames when there is no key to match on.
Python
orders.merge(
customers,
on="customer_id",
how="left",
validate="many_to_one",
)Related practice topics
Track your streak & earn XP
Sign up free to unlock your progress dashboard, daily streaks, and leaderboard ranking.
Full-Length Interview Practice Tests100% Free
10 role-based tests for Data Analysts, SQL Developers & Data Engineers with instant scorecards.