我有一个带有 DateTimeIndex 的 Pandas 数据框和一个名为 WEEKEND 的空列。
如果索引中的日期时间是周末,我想将该列的值设置为“YES”,以便生成的数据帧如下所示:
TIME WEEKEND
2019-01-01 00:00:00 NO
2019-01-02 06:00:00 NO
2019-01-03 21:00:00 NO
2019-01-04 18:00:00 NO
2019-01-05 03:00:00 YES
2019-01-06 07:00:00 YES
2019-01-07 10:00:00 NO
2019-01-08 08:00:00 NO
我知道我能做到:
df.loc[df.index.dayofweek == 5, 'WEEKEND'] = "YES" # Saturday
df.loc[df.index.dayofweek == 6, 'WEEKEND'] = "YES" # Sunday
但我想在一个语句中完成此操作,可能使用 IN 关键字。
我尝试了这个,但出现错误:
df.loc[df.index.dayofweek in [5,6], 'WEEKEND'] = "YES"
实现这一目标的最佳方法是什么?
谢谢
最佳答案
您可以在 .loc 中申请或
df.loc[(df.index.dayofweek == 5) | (df.index.dayofweek == 6), 'WEEKEND'] = "YES"
使用 .isin 的第二种解决方案
df.loc[df.index.dayofweek.isin([5,6]) , 'WEEKEND'] = "YES"
关于python - 根据 DateTimeIndex 更新列,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/57061006/