使用变量过滤pandas数据帧中的行

import pandas as pd

# 创建一个示例DataFrame
data = {'city': ['Beijing', 'Shanghai', 'Guangzhou'],
        'temperature': [32, 31, 28]}
df = pd.DataFrame(data)

# 使用布尔索引过滤行
condition = df['temperature'] >= 30
filtered_df = df[condition]

print(filtered_df)

       city  temperature
0   Beijing           32
1  Shanghai           31

import pandas as pd

# 创建一个示例DataFrame
data = {'city': ['Beijing', 'Shanghai', 'Guangzhou'],
        'temperature': [32, 31, 28]}
df = pd.DataFrame(data)

# 使用query()方法过滤行
temperature_threshold = 30
filtered_df = df.query("temperature >= @temperature_threshold")

print(filtered_df)