如何根据变量为Pandas DataFrame列赋值0或1？

你可以用Python的三元表达式直接一行完成赋值，写法更紧凑：

variable = 5
df['column name'] = 1 if variable == 5 else 0

还有更简洁的方式：利用Python中布尔值与整数的转换特性（True等价于1，False等价于0），直接将比较结果转成整数：

variable = 5
df['column name'] = int(variable == 5)

这两种写法都能替代原来的多行if-else逻辑，代码更简洁易读。

内容的提问来源于stack exchange，提问作者cs0815