анных нижеприведенного кадра:

x  y  a b c 
2  6 12 1 2
1  2  4 6 8

Я хочу привести к новому столбцу (т.е. d), который возвращает имя столбца с максимальным значением только среди a, b, c.

cols
a
c

Я пытаюсь найти максимальные значения из трех столбцов и вернуть имя столбца. Но вместо того, чтобы выбрать все строки набора данных, я хочу выбрать строки только из этих трех столбцов. Я использую следующий код:

def returncolname(row, colnames):
    return colnames[np.argmax(row.values)]
data['colmax'] = data.apply(lambda x: returncolname(x, data.columns), axis=1)