在组内用Python创建新列

Question

我需要创建一个新变量，如下面Spyder中给出的快照所示。我出错了。请给出解决方案。

enter image description here

提前谢谢！！！！

Answer 1

您可以执行以下操作：

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)