Friday, January 3, 2025

Pandas .query() method

import pandas as pd


data = {'first name': ['Alice', 'Bob', 'Charlie', 'David'],

        'age': [25, 30, 35, 40],

        'salary': [50000, 60000, 70000, 80000]}

df = pd.DataFrame(data)


filter1 = df[(df['age'] > 30) | (df['salary'] > 60000)]

print(filter1, '\n')


filter2 = df.query('age > 30 or salary > 60000')

print(filter2, '\n')


# Using variables

Age = 30

filter3 = df.query("age > @Age")

print(filter3, '\n')


# Using backtick when column name contains space

filter4 = df.query("`first name` == 'Charlie'")

print(filter4)


No comments:

Post a Comment