как применить функцию к столбцу pandas
Ответы
Ivan Mamtsev
29 мая 2024
Для применения функции к столбцу pandas можно использовать метод apply
. Этот метод применяет указанную функцию к каждому элементу столбца и возвращает новый столбец с результатами применения этой функции.
Пример:
import pandas as pd
# создание DataFrame
data = {'A': [1, 2, 3, 4],
'B': [5, 6, 7, 8]}
df = pd.DataFrame(data)
# создание функции, которую мы хотим применить к столбцу
def add_one(x):
return x + 1
# применение функции к столбцу 'A' и сохранение результатов в новом столбце 'A_plus_1'
df['A_plus_1'] = df['A'].apply(add_one)
Этот код создаст новый столбец 'A_plus_1', в котором каждому элементу столбца 'A' будет добавлена единица.
Также можно использовать лямбда-функции для более простых операций:
# применение лямбда-функции к столбцу 'B' и сохранение результатов в новом столбце 'B_times_2'
df['B_times_2'] = df['B'].apply(lambda x: x * 2)
Таким образом, метод apply
позволяет применить функцию к столбцу pandas и сохранить результаты операции в новом столбце.
0
0