是指在一个数据框(DataFrame)中添加一个列,该列记录每一行数据是否是工作日。工作日一般指周一至周五,不包括法定假日。
实现这个功能的步骤如下:
import pandas as pd
import numpy as np
df = pd.DataFrame({'日期': ['2022-01-01', '2022-01-02', '2022-01-03', '2022-01-04', '2022-01-05', '2022-01-06']})
df['日期'] = pd.to_datetime(df['日期'])
df['工作日'] = np.where(df['日期'].dt.weekday < 5, True, False)
在这个例子中,使用了NumPy库的np.where()
函数来根据条件判断工作日是否为True(工作日)或False(非工作日)。df['日期'].dt.weekday
返回的是日期对应的星期几,星期一到星期五对应的值分别为0到4。
最后,我们可以打印输出数据框来查看结果:
print(df)
输出结果:
日期 工作日
0 2022-01-01 False
1 2022-01-02 False
2 2022-01-03 True
3 2022-01-04 True
4 2022-01-05 True
5 2022-01-06 True
这个方法的优势是简单易懂,适用于处理日期数据和添加工作日列的常见需求。
腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云