python - 将数据框日期行转换为周末/非周末值

Question

我有一个带有 DATE 行的数据框，如果日期是周末，我需要将其转换为值 1 的行，如果不是，则为 0。

到目前为止，我将数据转换为工作日

df['WEEKDAY'] = pandas.to_datetime(df['DATE']).dt.dayofweek

有没有办法在没有函数的情况下创建这个“WEEKEND”行？

谢谢!

Answer 1

这是我想出的解决方案:

df['WEEKDAY'] = ((pd.DatetimeIndex(df.index).dayofweek) // 5 == 1).astype(float)

基本上它所做的就是使用整数除法 (//) 来测试 DatetimeIndex 的 dayofweek 属性是否小于 5。通常这将只返回一个 True 或 False，但是在最后加上 astype(float) 返回 1 或 0 而不是 bool 值。