【Python】进阶学习:pandas--如何根据指定条件筛选数据
在Pandas中,您可以使用.loc[]
或.query()
方法来根据指定条件筛选数据。以下是两种方法的示例代码:
使用.loc[]
方法:
import pandas as pd
# 创建示例DataFrame
df = pd.DataFrame({
'A': [1, 2, 3, 4],
'B': [5, 6, 7, 8],
'C': [9, 10, 11, 12]
})
# 指定条件筛选数据
condition = df['A'] > 2
filtered_df = df.loc[condition]
print(filtered_df)
使用.query()
方法:
import pandas as pd
# 创建示例DataFrame
df = pd.DataFrame({
'A': [1, 2, 3, 4],
'B': [5, 6, 7, 8],
'C': [9, 10, 11, 12]
})
# 使用query()方法筛选数据
filtered_df = df.query('A > 2')
print(filtered_df)
两种方法都可以根据条件筛选出满足条件的数据行。.loc[]
接受一个布尔数组作为索引,而.query()
可以直接使用查询字符串表示条件。
评论已关闭