python - 如何在 python 中创建条件列

Question

如何为条件语句创建 python 语句

我有一个类似的数据框

Day   Weather   Indicator
1     Sunny     No 
2     Sunny     No
3     Dark      No
7     Dark      No

我如何创建一个条件，让我知道如果 day=7 然后创建一个新列

Day   Weather   Indicator
1     Sunny     No 
2     Sunny     No
3     Dark      No
7     Dark      Yes

Answer 1

使用numpy.where :

df['Indicator'] = np.where(df['Day'] == 7, 'Yes','No')

编辑:

如果存在 Indicator 列，则填充 No 值 - mask或 loc按条件选择行的工作很好:

df['Indicator'] = df['Indicator'].mask(df['Day'] == 7, 'Yes')

---

df.loc[df['Day'] == 7, 'Indicator'] = 'Yes'

print (df)
   Day Weather Indicator
0    1   Sunny        No
1    2   Sunny        No
2    3    Dark        No
3    7    Dark       Yes