Conditional selection with .loc in Pandas DataFrame

Last update on December 21 2024 07:40:53 (UTC/GMT +8 hours)

Pandas: Advanced Indexing and Slicing Exercise-10 with Solution

Conditional Selection with loc:

Write a Pandas program to use .loc to select rows based on a condition.

Sample Solution :

Python Code :

import pandas as pd

# Create a DataFrame
df = pd.DataFrame({
    'P': [1, 6, 8, 3, 7],
    'Q': [5, 2, 9, 4, 1]
})

# Conditional selection using .loc
result = df.loc[df['P'] > 3]
print(result)

Output:

   P  Q
1  6  2
2  8  9
4  7  1

Explanation:

• Import pandas library.
• Create a DataFrame.
• Use .loc to select rows where column 'P' > 3.
• Print the results.

Python-Pandas Code Editor:

Have another way to solve this solution? Contribute your code (and comments) through Disqus.

Previous: Selecting rows by position in Pandas DataFrame.
Next: Setting values with .loc in Pandas DataFrame.