pandas как присвоить значение по условию

Аватар пользователя Maksim Litvinov
Maksim Litvinov
10 июля 2024

Для присвоения значения по условию в pandas можно использовать метод loc[], который позволяет обращаться к строкам и столбцам по меткам индекса и названиям столбцов.

Например, если мы хотим присвоить значение "Новое значение" столбцу 'Колонка2' в строках, где значение столбца 'Колонка1' равно 10, можно сделать следующим образом:

import pandas as pd

data = {'Колонка1': [10, 20, 30],
        'Колонка2': ['Значение1', 'Значение2', 'Значение3']}

df = pd.DataFrame(data)

# Присваиваем новое значение в столбец 'Колонка2' по условию
df.loc[df['Колонка1'] == 10, 'Колонка2'] = 'Новое значение'

print(df)

Этот код присвоит значение "Новое значение" столбцу 'Колонка2' только в строке, где значение столбца 'Колонка1' равно 10.

1 0