как применить функцию к столбцу pandas

Аватар пользователя Ivan Mamtsev
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
Познакомьтесь с основами аналитики данных бесплатно