创建pandas数据如下:
import pandas as pd# 创建数据
df = pd.DataFrame({'A': [1, 2, 3],'B': [4, 5, 6],'C': [7, 8, 9]
})
若进行数据查询,则可有以下几种查询方式:
1、使用布尔索引(简单便捷,推荐使用)
# 使用布尔索引选择数据
filtered_df = df[(df['A'] > 1) & (df['B'] < 6)]
print(filtered_df)
2、使用query
方法(简单便捷,推荐使用)
# 使用query方法选择数据
filtered_df = df.query('A > 1 and B < 6')
print(filtered_df)
3、 使用isin
方法进行成员资格查询(简单便捷,推荐使用)
# 使用isin方法选择数据
filtered_df = df[df['A'].isin([1, 3])]
print(filtered_df)
4、使用loc
和iloc
结合条件表达式(不推荐使用)
# 使用loc选择数据
filtered_df = df.loc[(df['A'] > 1) & (df['B'] < 6)]
print(filtered_df)
5、 使用apply
方法结合自定义函数(不推荐使用)
# 使用apply方法选择数据,这里定义一个函数来检查条件
def my_filter(row):return row['A'] > 1 and row['B'] < 6filtered_df = df[df.apply(my_filter, axis=1)]
print(filtered_df)