用于获取某个类别的先前日期的新列(Python)

import datetime

# 获取当前日期
current_date = datetime.date.today()

# 获取指定类别的先前日期
def get_previous_date(category):
    if category == 'week':  # 上一周的日期
        previous_date = current_date - datetime.timedelta(days=7)
    elif category == 'month':  # 上一个月的日期
        previous_date = current_date - datetime.timedelta(days=30)
    elif category == 'year':  # 上一年的日期
        previous_date = current_date - datetime.timedelta(days=365)
    else:
        previous_date = None  # 非法类别

    return previous_date

# 测试例子
print(get_previous_date('week'))  # 获取上一周的日期
print(get_previous_date('month'))  # 获取上一个月的日期
print(get_previous_date('year'))  # 获取上一年的日期