Панды добавьте столбец со значением на основе условия на основе других столбцов ⇐ Python

[Anonymous]

У меня есть следующий Pandas DataFrame:

Код: Выделить всё

import pandas as pd
import numpy as np

d = {'age' : [21, 45, 45, 5],
'salary' : [20, 40, 10, 100]}

df = pd.DataFrame(d)

и хотел бы добавить дополнительный столбец под названием «is_rich», который захватывает, если человек богат в зависимости от его/ее зарплаты. Я нашел несколько способов выполнить это: < /p>

Код: Выделить всё

# method 1
df['is_rich_method1'] = np.where(df['salary']>=50, 'yes', 'no')

# method 2
df['is_rich_method2'] = ['yes' if x >= 50 else 'no' for x in df['salary']]

# method 3
df['is_rich_method3'] = 'no'
df.loc[df['salary'] > 50,'is_rich_method3'] = 'yes'

, в результате чего:

Однако я не понимаю, что является предпочтительным способом. Все ли методы одинаково хороши в зависимости от вашего приложения?

Подробнее здесь: https://stackoverflow.com/questions/503 ... er-columns