从按ID的客户购买日期到python中按ID的购买间隔天数

import datetime

# 假设有一个包含客户ID和购买日期的字典
customer_data = {
    '001': '2022-01-01',
    '002': '2022-01-05',
    '003': '2022-01-10',
    '004': '2022-01-15',
}

# 将购买日期转换为日期对象并按ID排序
sorted_customer_data = sorted(customer_data.items(), key=lambda x: x[0])
sorted_customer_data = [(customer_id, datetime.datetime.strptime(date_str, '%Y-%m-%d')) for customer_id, date_str in sorted_customer_data]

# 计算购买间隔天数
purchase_intervals = {}
for i in range(1, len(sorted_customer_data)):
    customer_id, prev_date = sorted_customer_data[i-1]
    _, curr_date = sorted_customer_data[i]
    interval = (curr_date - prev_date).days
    purchase_intervals[customer_id] = interval

# 打印购买间隔天数
for customer_id, interval in purchase_intervals.items():
    print(f"客户ID: {customer_id}，购买间隔天数: {interval}")