我想将列表中的列名提取到按每行值过滤的Series
In [1]: import pandas as pd
In [2]: df =pd.DataFrame({'colA':[1,0,1], 'colB':[0,0,1], 'colC':[1,0,0]})
In [3]: print(df)
colA colB colC
0 1 0 1
1 0 0 0
2 1 1 0
生成的系列应如下所示：
0 [colA, colC]
1 []
2 [colA, colB]
dtype: object
这是我想出的折磨解决方案：
In [4]: df2 = df.T
In [5]: l = [df2[df2[i]>0].index.values.tolist() for i in range(3)]
In [6]: print(pd.Series(l))
0 [colA, colC]
1 []
2 [colA, colB]
dtype: object
有没有那么折磨的方式呢？
最佳答案
您可以使用np.where假定您的数据帧由0和1组成，并根据结果创建一个Series：
x = np.where(df,df.columns,'')
pd.Series([' '.join(i).split() for i in x])
0 [colA, colC]
1 []
2 [colA, colB]
关于python - 从DataFrame中按行提取列名称到Series，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/53704052/