python - 有什么方法可以根据 Pandas 中的特定条件在数据框的所有行中添加列名？

Question

我有一个像这样的数据框:

   Name  Age  Class Maths English Physics Bio Chemistry
    A     13   7      1    None     None   1    None
    B     17   10    None   1        1    None  None

我想添加一个名为 Subject 的新列，该列应包括列名称(具有 1)作为主题字段的值，如下所示:

   Name  Age  Class    Subject     
    A     13   7     Maths, Bio
    B     17   10    English, Physics

我尝试了几种方法，但比平时花费的时间更长。

Answer 1

您可以使用 apply使用 lambda 函数。

df['Subject'] = (df == '1').apply(lambda x: ','.join(df.columns[x]), axis=1)
df = df.iloc[:, [0,1,2,-1]]
df
  Name  Age  Class           Subject
0    A   13      7        Maths, Bio
1    B   17     10  English, Physics