Python实现客户列表按月年分批与分配：高效数据组织策略

months = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
years = ['2024', '2025', '2026', '2027', '2028', '2029', '2030', '2031', '2032']

# 正确生成按月年顺序排列的时间序列
# 确保先遍历年份，再遍历月份，以得到 Jan-2024, Feb-2024, ..., Dec-2024, Jan-2025 的顺序
month_years = [f"{m}-{y}" for y in years for m in months]

print(month_years[:15]) # 打印前15个，验证顺序
# 预期输出: ['Jan-2024', 'Feb-2024', 'Mar-2024', 'Apr-2024', 'May-2024', 'Jun-2024', 'Jul-2024', 'Aug-2024', 'Sep-2024', 'Oct-2024', 'Nov-2024', 'Dec-2024', 'Jan-2025', 'Feb-2025', 'Mar-2025']

# 假设有一个客户邮件列表
all_customers = [f'email{i+1}@example.com' for i in range(5000)] # 示例：5000个客户

chunk_size = 500 # 每批客户数量

# 将客户列表按指定大小分块
customer_chunks = [all_customers[i:i + chunk_size] for i in range(0, len(all_customers), chunk_size)]

print(f"分块数量: {len(customer_chunks)}")
print(f"第一个分块: {customer_chunks[0][:5]}...") # 打印第一个分块的前5个客户
# 预期输出: 分块数量: 10, 第一个分块: ['email1@example.com', 'email2@example.com', 'email3@example.com', 'email4@example.com', 'email5@example.com']...

def generate_customer_bookings_by_month_year(customer_list: list, chunk_size: int, months: list, years: list) -> dict:
    """
    将客户列表按指定大小分块，并按月-年时间序列进行分配。

    Args:
        customer_list (list): 包含所有客户的原始列表。
        chunk_size (int): 每个时间周期分配的客户数量。
        months (list): 月份名称列表 (e.g., ['Jan', 'Feb', ...])。
        years (list): 年份列表 (e.g., ['2024', '2025', ...])。

    Returns:
        dict: 键为 '月-年' 格式的时间周期，值为对应的客户列表分块。

    Raises:
        AssertionError: 如果生成的时间周期数量不足以分配所有客户分块。
    """

    # 1. 生成按月年顺序排列的时间序列
    month_years = [f"{m}-{y}" for y in years for m in months]

    # 2. 将客户列表按指定大小分块
    customer_chunks = [customer_list[i:i + chunk_size] for i in range(0, len(customer_list), chunk_size)]

    # 3. 验证时间序列数量是否足够
    assert len(customer_chunks) <= len(month_years), \
        f"错误：时间周期数量不足。需要 {len(customer_chunks)} 个时间周期，但只生成了 {len(month_years)} 个。"

    # 4. 将客户分块映射到时间序列，并返回字典
    return dict(zip(month_years, customer_chunks))

# --- 示例用法 ---

# 定义月份和年份
months_list = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
years_list = ['2024', '2025', '2026', '2027', '2028', '2029', '2030', '2031', '2032']

# 模拟一个包含50个客户的列表（便于演示，实际可为5000或更多）
dummy_customers = [f'email{i+1}@example.com' for i in range(50)]

# 设置每个时间周期的客户数量（演示用5个，实际可为500）
booking_chunk_size = 5

# 调用函数生成客户预约列表
customer_bookings = generate_customer_bookings_by_month_year(
    customer_list=dummy_customers,
    chunk_size=booking_chunk_size,
    months=months_list,
    years=years_list
)

# 打印结果
import json
print(json.dumps(customer_bookings, indent=4))

# 尝试一个时间周期不足的场景（这将触发 AssertionError）
# try:
#     generate_customer_bookings_by_month_year(
#         customer_list=[f'email{i+1}' for i in range(200)], # 200个客户
#         chunk_size=5, # 每批5个，需要40个时间周期
#         months=['Jan', 'Feb'], # 只有2个月
#         years=['2024'] # 只有1年，总共只有2个时间周期
#     )
# except AssertionError as e:
#     print(f"\n捕获到错误: {e}")

{
    "Jan-2024": [
        "email1@example.com",
        "email2@example.com",
        "email3@example.com",
        "email4@example.com",
        "email5@example.com"
    ],
    "Feb-2024": [
        "email6@example.com",
        "email7@example.com",
        "email8@example.com",
        "email9@example.com",
        "email10@example.com"
    ],
    "Mar-2024": [
        "email11@example.com",
        "email12@example.com",
        "email13@example.com",
        "email14@example.com",
        "email15@example.com"
    ],
    // ... 更多月份 ...
    "Oct-2024": [
        "email46@example.com",
        "email47@example.com",
        "email48@example.com",
        "email49@example.com",
        "email50@example.com"
    ]
}