Создать новый столбец в Python внутри группы - PullRequest
0 голосов
/ 12 января 2019

Мне нужно создать новую переменную, проиллюстрированную как «Требуется» в оснастке, приведенной ниже в Spyder. Я получаю ошибку. Пожалуйста, дайте решения.

введите описание изображения здесь

Заранее спасибо !!!!

1 Ответ

0 голосов
/ 14 января 2019

вы могли бы сделать что-то вроде этого:

def f(row):
  if (row['id'] == 1):
      return "Both"
  elif (row['id'] == 2):
      return "N"
  elif (row['id'] == 3):
      return "Y"

df["needed"] = df.apply(f,axis=1)
...