如何为Pandas DataFrame添加含符合条件列名列表的新列

示例数据构建 #

先还原你的DataFrame：

import pandas as pd

df = pd.DataFrame({
    'A': [2, 2, 3],
    'B': [4, 5, 6],
    'C': [1, 6, 2],
    'D': [7, 7, 7]
})

高效解决方案（适合列数多的场景） #

使用向量化的布尔矩阵点积操作，避免逐行/逐列的低效循环：

# 生成布尔矩阵，标记每个元素是否满足条件（>=3）
mask = df >= 3
# 通过点积快速拼接每行符合条件的列名，再转为列表
df['E'] = mask.dot(df.columns + ',').str.rstrip(',').str.split(',')

方法说明 #

1. mask = df >=3 生成一个和原DataFrame同形状的布尔矩阵，每个单元格为True表示对应元素满足条件；
2. mask.dot(df.columns + ',') 利用矩阵点积的特性，将每行中True对应的列名用逗号连接成字符串（比如第一行得到"B,D,"）；
3. .str.rstrip(',') 去掉字符串末尾多余的逗号，.str.split(',') 将字符串拆分为列表，最终得到你需要的列E。

最终效果 #

执行后你的DataFrame会变成：

这个方法是向量化操作，比np.where逐列判断、apply逐行处理的效率高得多，尤其适合列数多的场景。

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