在Python3上标记DataFrame中的假日,可以使用pandas库和holidays库来实现。
首先,确保已经安装了pandas和holidays库。可以使用以下命令进行安装:
pip install pandas
pip install holidays
接下来,导入所需的库:
import pandas as pd
import holidays
然后,创建一个包含日期的DataFrame:
data = {'date': ['2022-01-01', '2022-01-02', '2022-01-03', '2022-01-04']}
df = pd.DataFrame(data)
接下来,使用holidays库来获取假日列表:
holiday_list = holidays.CountryHoliday('US')
然后,创建一个函数来标记DataFrame中的假日:
def mark_holidays(row):
date = pd.to_datetime(row['date']).date()
if date in holiday_list:
return 'Holiday'
else:
return 'Not Holiday'
df['holiday'] = df.apply(mark_holidays, axis=1)
最后,打印标记后的DataFrame:
print(df)
输出结果将显示每个日期是否为假日:
date holiday
0 2022-01-01 Holiday
1 2022-01-02 Holiday
2 2022-01-03 Not Holiday
3 2022-01-04 Not Holiday
这样,你就可以在Python3上标记DataFrame中的假日了。
请注意,以上示例中使用的是美国的假日列表。如果需要其他国家或地区的假日列表,可以根据holidays库的文档进行相应的调整。
腾讯云相关产品和产品介绍链接地址:
领取专属 10元无门槛券
手把手带您无忧上云