如何在 Python Pandas 数据框中选择行

说明

1、布尔索引( df[df['col'] == value] )

2、位置索引( df.iloc[...])

3、标签索引( df.xs(...))

4、df.query(...)应用程序接口

下面显示了每种方法的示例，以及何时使用某些技术的建议。假设我们的标准是列 'A'=='foo'

（关于性能的注意事项：对于每个基本类型，我们可以使用 Pandas API 使事情变得简单，或者我们可以在 API 之外冒险，通常进入 NumPy，并加快速度。）

设置

我们需要做的第一件事是确定一个条件，它将作为我们选择行的标准。我们将从 OP 的案例 column_name == some_value 开始，并包括一些其他常见用例。

借用@unutbu：

import pandas as pd, numpy as np
df = pd.DataFrame({'A': 'foo bar foo bar foo bar foo foo'.split(),
                   'B': 'one one two three two two one three'.split(),
                   'C': np.arange(8), 'D': np.arange(8) * 2})

本文教程操作环境：windows7系统、Python 3.9.1，DELL G3电脑。