python 3根据条件获取列名

参见英文答案 > Create a column in a dataframe that is a string of characters summarizing data in other columns                                    3个
所以我有像这样的pandas df(python 3.6)

index   A   B   C  ... 
  A     1   5   0
  B     0   0   1 
  C     1   2   4
 ...

如您所见,索引值与列名称相同.

我要做的是在数据框中获取一个新列,其中列的名称为>比0

index   A   B   C  ... NewColumn
  A     1   5   0       [A,B]
  B     0   0   1       [C]
  C     1   2   4       [A,B,C]
 ...

我一直在尝试iterrows没有成功

我也知道我可以融化和转动,但我认为应该有一种方式适用lamnda可能吗？

提前致谢

解决方法:

如果新列应该是字符串比较DataFrame.gt与dot产品列,最后删除尾随分隔符：

df['NewColumn'] = df.gt(0).dot(df.columns + ', ').str.rstrip(', ')
print (df)
   A  B  C NewColumn
A  1  5  0      A, B
B  0  0  1         C
C  1  2  4   A, B, C

对于列表,使用apply with lambda function：

df['NewColumn'] = df.gt(0).apply(lambda x: x.index[x].tolist(), axis=1)
print (df)
   A  B  C  NewColumn
A  1  5  0     [A, B]
B  0  0  1        [C]
C  1  2  4  [A, B, C]