python - 根据 DateTimeIndex 更新列

Question

我有一个带有 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"

实现这一目标的最佳方法是什么？

谢谢

Answer 1

您可以在 .loc 中申请或

df.loc[(df.index.dayofweek == 5) | (df.index.dayofweek == 6), 'WEEKEND'] = "YES"

使用 .isin 的第二种解决方案

df.loc[df.index.dayofweek.isin([5,6]) , 'WEEKEND'] = "YES"