Поиск по индексу

Question

У меня есть столбец Search, который содержит строку заголовка согласно:

Alpha Bravo Charlie Search SearchReturn
1      2      3     Alpha      1  
2      5      6     Charlie    6

Я пытаюсь использовать что-то вроде функции Excel Vlookup для SearchReturn Столбец.

Однако я не знаю, как создать колонку SearchReturn, вы можете посоветовать?

Answer 1

Это сделает это:

df['SearchReturn'] = df.apply(lambda x: x[x['Search']], axis=1)

Answer 2

Используйте pd.DataFrame.lookup с метками строк и столбцов:

df['SearchReturn'] = df.lookup(df.index, df['Search'])

print(df)

   Alpha  Bravo  Charlie   Search  SearchReturn
0      1      2        3    Alpha             1
1      2      5        6  Charlie             6