pandas生成简单的日历表

最新推荐文章于 2023-09-13 22:59:53 发布

原创最新推荐文章于 2023-09-13 22:59:53 发布 · 294 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#pandas #python #数据分析

该代码段展示了如何使用Pandas对DataFrame中的日期列进行操作，提取如年、月、日等时间维度。通过`eval`函数将字符串转化为表达式，添加新的列到数据框中。之后使用`apply`函数结合`calendar_add`函数批量处理多个日期属性。最后，创建了一个包含多种日期特征的示例数据框`calendar`。

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()