def calendar_add(df: pd.DataFrame, dim_list: list):
''' 提取指定的日期维度 '''
for dim in dim_list:
# eval 把字符串变成表达式
df[dim] = eval(f"df['date'].{dim}")
return df
calendar = pd.DataFrame()
calendar['date'] = pd.date_range('20230328', '20231231')
# calendar['year'] = calendar['date'].dt.day_name()
dim_list = ['year', 'month', 'day', 'quarter',
'day_name()', 'weekofyear', 'dayofyear',
'daysinmonth', 'dayofweek', 'is_month_end',
'is_month_start', 'is_quarter_end', 'is_quarter_start'
]
# apply 多参数的用法注意
calendar = calendar.apply(calendar_add, args=(dim_list,), axis=1)
# type(calendar['date'])
calendar.head()
相关文字
python如何通过函数名的字符串来调用这个函数
日期时间列中提取月份和年份
将时间列转换为年-月-日
pandas之apply函数简介及用法