dataframe.apply () при сохранении имен столбцов

Question

Я хочу применить функцию к каждой строке в кадре данных. Я использую apply (), но затем он передает объект Series, поэтому имена столбцов не передаются. Есть ли способ использовать .apply () и сохранить имена столбцов?

Answer 1

, если вы хотите применить строку функции мудро:

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

# df

    name value
0   AA  1
1   BB  2
2   CC  3

def  function_2(row):
    return row['value'] * 2

df['col_b'] = df.apply(lambda row: function_2(row), axis=1)

вывод:

    name value col_b
0   AA  1   2
1   BB  2   4
2   CC  3   6

Пример ссылки на источник