如何用Python Pandas检查行中指定列是否含1并输出对应行名

步骤与代码示例 #

假设你的DataFrame索引就是示例中的名称（C、NC、C1等），可以通过以下代码实现需求：

import pandas as pd

# 构造示例数据（替换成你的实际DataFrame）
data = {'Braf': [1, 0, 1]}
df = pd.DataFrame(data, index=['C', 'NC', 'C1'])

# 筛选Braf值为1的行并提取索引
target_names = df[df['Braf'] == 1].index.tolist()

# 输出结果
print(target_names)  # 输出: ['C', 'C1']

代码说明 #

• df['Braf'] == 1：生成布尔序列，标记每行Braf列的值是否为1
• df[布尔序列]：筛选出符合条件的行
• .index.tolist()：将筛选结果的索引转换为列表，直接得到所需的名称列表

特殊情况处理 #

如果名称不是索引而是单独的列（比如列名为Name），只需调整提取逻辑：

target_names = df[df['Braf'] == 1]['Name'].tolist()

内容的提问来源于stack exchange，提问作者Bubbelz