一分钟实现python取出dataframe中某列值等于某几个固定值的所有信息

原创已于 2024-04-16 15:45:37 修改 · 1.6k 阅读

9 ·

CC 4.0 BY-SA版权

文章标签：

#python #pandas

于 2024-01-24 15:28:13 首次发布

本文介绍了如何在Python的pandas库中，利用isin()函数从DataFrame中筛选出QPRE列中值等于0、2或3的所有行，通过一个示例展示了实际操作过程。

部署运行你感兴趣的模型镜像

1、解释说明

在Python中，可以使用pandas库中的DataFrame来实现取出dataframe中某列值等于某几个固定值的所有信息。具体操作是使用isin()函数来筛选出符合条件的行。

2、使用示例

假设我们有一个名为data_pre_1min的DataFrame，其中有一列名为QPRE，我们想要筛选出QPRE列中值为0、2或3的所有行。可以使用以下代码实现：

import pandas as pd

# 创建一个示例DataFrame
data = {'A': [1, 2, 3, 4, 5],
        'B': [6, 7, 8, 9, 10],
        'QPRE': [0, 2, 3, 1, 2]}
data_pre_1min = pd.DataFrame(data)

# 筛选出QPRE列中值为0、2或3的所有行
filtered_data = data_pre_1min[data_pre_1min['QPRE'].isin([0, 2, 3])]
print(filtered_data)