如何获取数据框每一行中特定值的列名

Question

在一个数据帧中，我想检查当前行中的特定值，如果该值退出，我想获取该行中存在该值的列名，例如：

    Resource  Team  Mon Tue Wed Thu Fri   
19  Name1     Dev   S   S   L   L    S    

11  Name2     QA    L   W   S    L   S    

我想要在现有框架的新列中输出数据。 请告知我该如何实现。

预期的输出：

    Resource     OUTPUT
19  Name1        (S present in Mon,Tue,Fri L present in Wed, Thu)

11  Name2        (S present in Wed,Fri L present in Mon,Thu)

Answer 1

您可以创建一个可以沿axis=1应用的函数，然后将其应用到整个DataFrame。

def check_if_s_in_row(row):
    present = []
    for i in len(range(row)):
        if row[i] == "S":
            present.append(row.columns.values[i])
    return ["S present in {}".format(day) for day in present]

dataframe.apply(check_if_s_in_row, axis=1)

对L做适当的事。

Answer 2

据我了解，您可以执行以下操作：

m=df.set_index(['Resource','Team'])
m['S_present']=m.eq('S').dot(','+m.columns).str.lstrip(',')
m['L_present']=m.eq('L').dot(','+m.columns).str.lstrip(',')
print(m.reset_index())

---

  Resource Team Mon Tue Wed Thu Fri    S_present L_present
0    Name1  Dev   S   S   L   L   S  Mon,Tue,Fri   Wed,Thu
1    Name2   QA   L   W   S   L   S      Wed,Fri   Mon,Thu

Answer 3

这仅用于匹配您的输出

[' '.join(y.reset_index().groupby(x)['index'].apply(','.join).reset_index().apply(' present '.join,1))for x,y in df.iloc[:,2:].iterrows()]
Out[237]: 
['L present Wed,Thu S present Mon,Tue,Fri',
 'L present Mon,Thu S present Wed,Fri W present Tue']