python - 从列名构建列表 Pandas DataFrame

Question

所以我正在对相当大的数据集进行操作。我使用 Pandas DataFrame 来处理这些数据，并且坚持使用一种有效的方式将数据解析为两个格式化列表

这是我的数据框对象

            fet1    fet2    fet3    fet4    fet5
stim1       True    True    False   False   False
stim2       True    False   False   False   True
stim3       ...................................
stim4       ...................................
stim5       ............................. so on

我正在尝试解析每一行并创建两个列表。列表一应该有所有真值的列名。列表二应该有错误值的列名。

stim 1 的例子:

list_1=[fet1,fet2]   
list_2=[fet3,fet4,fet5]

我知道我可以强行使用这种方法并遍历行。或者我可以转置并转换为字典并以这种方式解析。我还可以创建稀疏序列对象，然后创建集合，但必须分别引用列名。

我遇到的唯一问题是我总是得到二次 O(n^2) 运行时间。

作为 Pandas 的内置功能，是否有更有效的方法来做到这一点？

感谢您的帮助。

Answer 1

这是你想要的吗？

>>> df
       fet1   fet2   fet3   fet4   fet5
stim1  True   True  False  False  False
stim2  True   False False  False   True
>>> def func(row):
        return [
            row.index[row == True], 
            row.index[row == False]
        ]
>>> df.apply(func, axis=1)
stim1    [[fet1, fet2], [fet3, fet4, fet5]]
stim2    [[fet1, fet5], [fet2, fet3, fet4]]
dtype: object