Создание динамического столбца Pandas

Question

Я пытаюсь динамически создать новый столбец на основе значений другого столбца. Скажем, у меня есть следующие dataframe

A | B
11 | 1
22 | 0
33 | 1
44 | 1
55 | 0

Я хочу создать новый колонка. Если значение столбца B равно 1, вставьте 'Y', иначе вставьте 'N'. Полученный dataframe должно выглядеть так:

A | B | C
11 | 1 | Y
22 | 0 | N
33 | 1 | Y
44 | 1 | Y
55 | 0 | N

Я мог бы сделать это перебором значений столбцов,

for i in dataframe['B'].values: 
    if i==1: 
     add Y to Series 
    else: 
     add N to Series 

dataframe['C'] = Series 

Howe Я боюсь, что это сильно снизит производительность, особенно потому, что мой набор данных содержит 500 000 строк. Любая помощь будет принята с благодарностью. Спасибо.

Answer 1

Избегайте связывания индексации с помощью loc. Есть некоторые тонкости с returning a view versus a copy в панд, которые связаны с numpy

df['C'] = 'N' 
df.loc[df.B == 1, 'C'] = 'Y' 

Answer 2

Попробуйте это:

df['C'] = 'N' 
df['C'][df['B']==1] = 'Y' 

должен быть быстрее.