Python Pandas 获取包含特定值的 DataFrame 行名
假设数据集是一个 pandas 的 DataFrame 对象,可以使用以下代码获取含有特定值的行名:
import pandas as pd
# 创建一个示例 DataFrame 对象
data = {'A': ['a', 'b', 'c', 'd', 'e'], 'B': [1, 2, 3, 4, 5]}
df = pd.DataFrame(data)
# 获取含有'a'的行名
rows_with_a = df[df['A'] == 'a'].index.tolist()
print(rows_with_a)
输出:
[0]
在上述代码中,使用了 pandas 的条件查询功能,将含有特定值的行筛选出来,并通过index.tolist()方法获取行名列表。如果有多行含有特定值,列表中会包含多个行名。
原文地址: https://www.cveoy.top/t/topic/oiHs 著作权归作者所有。请勿转载和采集!