Python / Pandas условно новый столбец ValueError

Question

Я хочу добавить новый столбец в существующий DataFrame и установить для df.TaxStatus значение True или False в зависимости от значения df.Public (1 | 2).Я вижу, как другие создают новый столбец, используя Numpy.where(), но я получаю ValueError.

df['TaxStatus'] = np.where(df.Public == 1, True, np.where(df.Public == 2, False))

Это может работать, за исключением ошибки ValueError: either both or neither of x and y should be given

Answer 1

Вы можете сделать это, просто сказав:

df['TaxStatus'] = df.Public == 1

Выражение df.Public == 1 возвращает серию значений True / False, которые вы можете назначить новому столбцу.Нет необходимости в np.where()