如何在Python中以切割段数为参数实现时段全切割可能性的函数？

原代码通过多层嵌套循环硬编码了将时段分割为5段的逻辑，一旦需要调整分割段数，就得修改循环结构，扩展性极差。下面提供一个通用函数，只需传入时段总长度和目标分割段数，就能生成所有合法的分割组合：

实现代码 #

import pandas as pd
import numpy as np
import itertools

def generate_period_splits(period_size, num_parts):
    if num_parts < 1:
        raise ValueError("分割段数必须大于等于1")
    if num_parts == 1:
        return pd.DataFrame({'period_array': [np.array([[0, period_size]])]})
    
    # 生成所有合法的n-1个分割点组合（分割点在1到period_size-1之间，天然升序）
    split_point_combinations = itertools.combinations(range(1, period_size), num_parts - 1)
    part_records = []
    
    for points in split_point_combinations:
        intervals = []
        start = 0
        for point in points:
            intervals.append([start, point])
            start = point
        intervals.append([start, period_size])
        part_records.append({'period_array': np.array(intervals)})
    
    return pd.DataFrame(part_records)

函数说明 #

• 核心逻辑：利用itertools.combinations生成所有符合要求的分割点组合——要分割成num_parts段，只需要选出num_parts-1个不重复的分割点（必须在1到period_size-1之间），这些组合天然按升序排列，刚好符合时段分割的顺序要求。
• 区间生成：遍历每个分割点组合，从0开始依次连接分割点，最终形成完整的时段区间序列，转换成numpy数组后存入结果列表。
• 边界处理：单独处理了分割段数为1的特殊情况，直接返回包含整个时段的结果。

使用示例 #

比如要将长度为6的时段分割为5段，调用方式如下：

result = generate_period_splits(6, 5)
print(result)

输出结果和原代码在period_size=6时的输出完全一致。

内容的提问来源于stack exchange，提问作者Shanki