如何将列表或Numpy数组正确划分为指定数量的分组（支持浮点数）

一、实现支持浮点数的等大小序列分组 #

你的第一个方法依赖整数除法和手动维护区间，确实会被限制在整数范围内，而且容易出现逻辑偏差。要实现数学上的等长度分组（支持浮点数），核心思路是基于空间均匀划分，而非按整数个数硬拆。

这里提供一个通用函数，能根据输入的区间起止值和分组数，生成等长度的子分组（支持浮点数），还能重复对任意子分组再次划分：

import numpy as np

def split_into_equal_groups(start: float, end: float, num_groups: int, num_points: int = None) -> list:
    """
    将区间[start, end]划分为num_groups个等长度的子分组
    :param start: 区间起始值
    :param end: 区间结束值
    :param num_groups: 分组数量
    :param num_points: 每个分组内生成的点数，默认和原区间整数个数成比例
    :return: 每个分组的数值序列列表
    """
    # 计算每个子区间的长度（支持浮点数）
    interval_length = (end - start) / num_groups
    # 如果没指定每个分组的点数，默认按原区间整数个数的比例分配
    if num_points is None:
        total_int_points = end - start + 1  # 原区间整数的数量
        num_points = int(np.round(total_int_points / num_groups))
    
    groups = []
    current_start = start
    for _ in range(num_groups):
        current_end = current_start + interval_length
        # 用linspace生成该子区间内均匀分布的数值（支持浮点数）
        group = np.linspace(current_start, current_end, num=num_points).tolist()
        groups.append(group)
        current_start = current_end
    
    return groups

# 测试第一次划分：1-22分成2组
first_groups = split_into_equal_groups(1, 22, 2)
print("第一次划分结果:\n", first_groups)

# 选择第一组，再次分成2组
selected_start = first_groups[0][0]
selected_end = first_groups[0][-1]
second_groups = split_into_equal_groups(selected_start, selected_end, 2)
print("\n对第一组再次划分结果:\n", second_groups)

这个方法的优势：

• 用np.linspace实现真正的等长度区间划分，完美支持浮点数
• 可以灵活控制每个分组内的点数，既可以按整数比例分配，也可以手动指定
• 不会修改原分组数据，每次划分都是基于选中的区间重新生成，避免数据丢失

二、修复Numpy数组方法中的类型错误 #

你的Numpy代码报错TypeError: 'tuple' object cannot be interpreted as an integer，主要是两个低级错误：

1. 错误地把tuple传给了range()，而range()只接受整数参数
2. 试图用函数调用的方式lis(np.arange(...))给数组添加元素，这不符合Numpy的语法

修复后的Numpy版本代码，同样实现等长度分组：

import numpy as np

begin = 1.0  # 用浮点数开头，确保后续计算都是浮点数
end = 22.0
num_groups = 2

# 计算每个子区间的长度（浮点数）
interval_length = (end - begin) / num_groups
# 每个分组内的点数，按原区间整数个数比例分配
num_points_per_group = int(np.round((end - begin + 1) / num_groups))

groups = []
current_start = begin
for _ in range(num_groups):
    current_end = current_start + interval_length
    # 用linspace生成均匀分布的数值序列（支持浮点数）
    group = np.linspace(current_start, current_end, num=num_points_per_group)
    groups.append(group)
    current_start = current_end

print("Numpy分组结果:\n", groups)

错误原因详解：

• 你原来的np.arange(range((begin, end+1)))中，range((begin, end+1))是把(begin, end+1)这个tuple传给range，range需要的是整数起止值，所以直接报错
• Numpy数组不能用lis(...)的方式添加元素，正确的做法是用列表存储每个分组的Numpy数组，或者用np.vstack等方法拼接，但这里用列表存更直观

内容的提问来源于stack exchange，提问作者steven-14