Pandas条件查询和过滤

2023-02-24 00:00:00 查询 条件 过滤

Pandas 是一种流行的 Python 数据分析工具,其中包括了很多方便的数据查询和过滤的函数。

要进行条件查询,可以使用 loc 或 iloc 函数。loc 函数用于基于标签的索引,而 iloc 函数用于基于位置的索引。以下是一些常见的条件查询示例:

1、根据单个条件查询:

import pandas as pd

# 创建数据帧
df = pd.DataFrame({'A': [1, 2, 3, 4, 5],
                   'B': [10, 20, 30, 40, 50]})

# 根据单个条件查询
result = df.loc[df['A'] > 3]
print(result)

输出:

   A   B
3  4  40
4  5  50

2、根据多个条件查询:

# 根据多个条件查询
result = df.loc[(df['A'] > 2) & (df['B'] < 40)]
print(result)

输出:

   A   B
2  3  30

3、选择指定列:

# 选择指定列
result = df.loc[df['A'] > 3, ['B']]
print(result)

输出:

    B
3  40
4  50

除了 loc 和 iloc 函数之外,Pandas 还提供了一些其他的函数,例如 query() 和 where(),可以用于进行更复杂的查询和过滤操作。

相关文章