Pandas DataFrame 中根据条件获取元素所在位置的索引方法

小石头
小石头
发布时间:2026-10-13| 3人浏览过
原创

本文介绍在 Pandas DataFrame 中,如何根据特定条件筛选行并获取其在原数据框中的位置索引。通过布尔索引结合 index.tolist() 方法,可精准定位符合条件的行索引列表,适用于需要定位数据位置的场景。

在 DataFrame 数据结构中,有时需要依据特定条件筛选数据,并找出符合该条件的行在原数据框中的具体位置索引。

实现该功能的核心代码如下:

df = pd.DataFrame({'BoolCol': [1, 2, 3, 3, 4],'attr': [22, 33, 22, 44, 66]}, 
  index=[10,20,30,40,50]) 
print(df) 
a = df[(df.BoolCol==3)&(df.attr==22)].index.tolist() 
print(a) 

假设存在如下 DataFrame 对象 df,通过筛选“BoolCol”列值为 3 且“attr”列值为 22 的行,可以获取该行在 df 中的索引位置。

需要注意的是,返回的结果是一个包含索引值的列表。由于 DataFrame 的索引(index)可能与默认的数字下标不同,因此返回的是实际的索引值,而非简单的序列号。

 BoolCol attr 
10  1 22 
20  2 33 
30  3 22 
40  3 44 
50  4 66 
[30] 

以上即为在 Pandas DataFrame 中根据条件获取元素所在位置(索引)的方法介绍,希望能为读者提供有效的参考。

免责声明:本文内容仅供参考,不代表本站观点。转载或引用请注明出处。

Python 相关产品和应用

全部下载
Selenium

Selenium

4.51.0 · Windows

用于浏览器自动化和 Web 测试的 Python 工具。

scikit-learn

scikit-learn

1.9.1 · Windows

常用 Python 机器学习算法与模型评估工具库。

Black

Black

26.10.1 · Windows

强调一致性的 Python 自动代码格式化工具。

Ruff

Ruff

0.17.1 · macOS

由 Rust 编写的高速 Python 代码检查和格式化工具。

pytest

pytest

9.1.1 · Windows

简洁而强大的 Python 自动化测试框架。

FastAPI

FastAPI

0.146.0 · Windows

高性能 Python API 框架,支持类型提示和自动接口文档。

Flask

Flask

3.1.3 · Windows

轻量灵活的 Python Web 应用框架。

Django

Django

6.1.2 · Windows

成熟的 Python Web 框架,适合构建内容站点和业务系统。

Spyder

Spyder

6.1.7 · Windows

面向科学计算与数据分析的 Python 集成开发环境。

SciPy

SciPy

1.18.1 · Windows

面向科学、工程和技术计算的 Python 算法库。