Вернуть имя столбца из индекса и значение с пандами

Question

Я ищу способ найти имя столбца с номером индекса и значением.

original = {'col1': [1, 2,0,9], 'col2': [3, 4, 2,5]}
original = pd.DataFrame(data=original)

    col1    col2
0   1        3
1   2        4
2   0        2
3   9        5

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

Я ожидаю вернуть только имя столбца col1.

Answer 1

Используйте loc для выбора строки и idxmax для выбора столбца:

(original.loc[1] == 2).idxmax()
# 'col1'

Если «1» - это целочисленная позиция, а не метка индекса, используйте iloc для выбора строки:

(original.loc[1] == 2).idxmax()
# 'col1'

Answer 2

В этом решении используется numpy.argwhere :

original.columns[np.argwhere(original.iloc[1,:] == 2)[0][0]]