如何在数据框中制作假期指标

Question

DataOrigin['isholiday'] = DataOrigin['arrival'].apply(lambda x: 1 if x=='2016-06-10'or'2016-06-09'or '2016-06-11'else 0)

enter image description here

请先单击图像，上面是我的代码 我的描述包括在图片中，表示赞赏！

Answer 1

您可以使用isin来检查值是否在给定的数据子集中。

DataOrigin['isholiday'] = DataOrigin['arrival'].isin(['2016-06-10', '2016-06-09', '2016-06-11'])

如果您真正想要的是1和0而不是True / False，则只需将.astype(int)附加到上面的语句中。正确/错误的评估结果应与1/0相同，因此实际上取决于偏好或您的特定用例。