Python教程：按月份和年份批量分割数据并进行时间关联

def chunk_list(data_list, chunk_size):
    """
    将一个列表分割成指定大小的子列表。

    Args:
        data_list (list): 待分割的原始列表。
        chunk_size (int): 每个子列表的大小。

    Returns:
        list: 包含所有子列表的列表。
    """
    return [data_list[i:i + chunk_size] for i in range(0, len(data_list), chunk_size)]

# 示例
dummy_data = [f'item_{i+1}' for i in range(20)]
chunked_data = chunk_list(dummy_data, 5)
print(chunked_data)
# 输出: [['item_1', 'item_2', 'item_3', 'item_4', 'item_5'], ..., ['item_16', 'item_17', 'item_18', 'item_19', 'item_20']]

def generate_month_year_sequence(months, years):
    """
    生成一个按月份和年份顺序排列的字符串序列。

    Args:
        months (list): 包含月份名称的列表，如 ['Jan', 'Feb', ...]。
        years (list): 包含年份的列表，如 ['2024', '2025', ...]。

    Returns:
        list: 格式为 'Month-Year' 的字符串列表。
    """
    # 关键在于循环的嵌套顺序：先遍历年份，再遍历月份，确保按时间顺序排列。
    return [f"{m}-{y}" for y in years for m in months]

# 示例
months = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
years = ['2024', '2025', '2026']
month_years_sequence = generate_month_year_sequence(months, years)
print(month_years_sequence[:5]) # 打印前5个，验证顺序
# 输出: ['Jan-2024', 'Feb-2024', 'Mar-2024', 'Apr-2024', 'May-2024']

def assign_chunks_to_periods(chunked_data, periods):
    """
    将分块数据与时间周期关联起来，并返回一个字典。

    Args:
        chunked_data (list): 包含分块子列表的列表。
        periods (list): 包含时间周期字符串的列表。

    Returns:
        dict: 键为时间周期，值为对应数据块的字典。
    """
    # 确保时间周期数量足够覆盖所有数据块
    if len(chunked_data) > len(periods):
        raise ValueError('时间周期数量不足以分配所有数据块。')

    return dict(zip(periods, chunked_data))

# 示例
# chunked_data 和 month_years_sequence 来自前面的示例
assigned_data = assign_chunks_to_periods(chunked_data, month_years_sequence)
print(assigned_data)
# 输出示例: {'Jan-2024': ['item_1', 'item_2', 'item_3', 'item_4', 'item_5'], ...}

import collections

def generate_monthly_customer_bookings(customer_emails, months, years, booking_limit_per_month):
    """
    将客户邮件列表按月份和年份分块，并为每个时间周期分配指定数量的客户。

    Args:
        customer_emails (list): 包含所有客户邮件地址的列表。
        months (list): 包含月份名称的列表，如 ['Jan', 'Feb', ...]。
        years (list): 包含年份的列表，如 ['2024', '2025', ...]。
        booking_limit_per_month (int): 每个月份-年份周期分配的客户数量上限。

    Returns:
        collections.OrderedDict: 键为 'Month-Year' 格式的时间周期，
                                 值为对应客户邮件列表的有序字典。
                                 使用有序字典可保持时间顺序。
    Raises:
        AssertionError: 如果生成的时间周期不足以分配所有客户数据块。
    """
    # 1. 生成有序的月份-年份序列
    # 确保先遍历年份，再遍历月份，以获得正确的 chronological order
    month_years = [f"{m}-{y}" for y in years for m in months]

    # 2. 将客户邮件列表分块
    # 使用列表推导式高效地将客户列表按指定大小分块
    customer_chunks = [
        customer_emails[i:i + booking_limit_per_month] 
        for i in range(0, len(customer_emails), booking_limit_per_month)
    ]

    # 3. 验证时间周期数量是否足够
    # 确保我们有足够多的 'Month-Year' 组合来分配所有的客户数据块
    assert len(customer_chunks) <= len(month_years), \
        f'错误：生成的时间周期 ({len(month_years)}) 不足以分配所有客户数据块 ({len(customer_chunks)})。'

    # 4. 将分块数据与时间周期关联
    # 使用 zip 函数将时间周期和客户数据块一一对应，并创建字典
    # 使用 OrderedDict 保持插入顺序，即时间顺序
    return collections.OrderedDict(zip(month_years, customer_chunks))

# --- 示例数据和调用 ---
# 定义月份和年份列表
months = ['Jan', 'Feb', 'Mar', 'Apr', 'May', 'Jun', 'Jul', 'Aug', 'Sep', 'Oct', 'Nov', 'Dec']
years = ['2024', '2025', '2026', '2027', '2028', '2029', '2030', '2031', '2032']

# 模拟5000个客户邮件地址
total_customers = 5000
l_lines = [f'customer{x+1}@example.com' for x in range(total_customers)]

# 每个月预订上限
booking_limit = 500

# 调用函数生成按月/年分配的客户列表
monthly_bookings = generate_monthly_customer_bookings(l_lines, months, years, booking_limit)

# 打印部分结果以验证
print("--- 前几个月的预订列表 ---")
for i, (month_year, customers) in enumerate(monthly_bookings.items()):
    if i < 5: # 打印前5个条目
        print(f"{month_year}: {customers[:3]}...") # 每个列表只打印前3个邮件
    else:
        break

print(f"\n总共分配了 {len(monthly_bookings)} 个时间周期。")
print(f"Jan-2024 的客户数量: {len(monthly_bookings['Jan-2024'])}")
print(f"Dec-2024 的客户数量: {len(monthly_bookings['Dec-2024'])}")

--- 前几个月的预订列表 ---
Jan-2024: ['customer1@example.com', 'customer2@example.com', 'customer3@example.com']...
Feb-2024: ['customer501@example.com', 'customer502@example.com', 'customer503@example.com']...
Mar-2024: ['customer1001@example.com', 'customer1002@example.com', 'customer1003@example.com']...
Apr-2024: ['customer1501@example.com', 'customer1502@example.com', 'customer1503@example.com']...
May-2024: ['customer2001@example.com', 'customer2002@example.com', 'customer2003@example.com']...

总共分配了 10 个时间周期。
Jan-2024 的客户数量: 500
Dec-2024 的客户数量: 500