我有一组1或0的列，具体取决于条件是否为真。每行只有一列是一列。
我正在尝试向df添加一个新列，该列找到值等于1 (e.g., column 0, 1, 2, etc.)的列号，并在与数据帧具有相同列元素数的列表中返回相应的索引值
例如，如果行1的第二列中的值为1，我希望新添加的列使用下面的列表作为参考读取b。< / p>
prim_cols = ['a','b','c','d','e','f']
用于此目的的最佳python函数是什么？我已经尝试了np.where，列表理解和其他一些，但我被卡住了。