case：如果holiday_date_list中包含“nextday_date”列中的值，我想在“nextday_holiday”中将其检查为1
下面的代码是正确的。但是这段代码可以更简短吗？
train = pd.read_csv("./train.csv")
holiday_date_list = [a, b, c, d]
idx = [idx for idx, value in enumerate(train["nextday_date"]) if value in holiday_date_list]
train.loc[idx, "nextday_holiday"] = 1
版权声明：本文内容由阿里云实名注册用户自发贡献，版权归原作者所有，阿里云开发者社区不拥有其著作权，亦不承担相应法律责任。具体规则请查看《阿里云开发者社区用户服务协议》和《阿里云开发者社区知识产权保护指引》。如果您发现本社区中有涉嫌抄袭的内容，填写侵权投诉表单进行举报，一经查实，本社区将立刻删除涉嫌侵权内容。
IIUC：
df = pd.DataFrame([["11","2", "6"], ["12","4", "2"], ["13","3", "4"]],
columns=["ix","a", "b"])
ix a b0 11 2 6
1 12 4 2
2 13 3 4
检查a中的值是否包含在b中的值中：
df['res']=df.a.isin(df.b).astype(int)
df
ix a b res0 11 2 6 1
1 12 4 2 1
2 13 3 4 0
评论
全部评论 (0)