基于N与V生成Python索引：输出不符合预期的问题求助

基于参数N和V生成索引时未得到预期输出，以下是实现代码、当前输出及期望输出：

实现代码 #

N = 3
V = [3, 4, 5, 6, 10, 11, 12, 13, 17, 18, 19, 20]

# 确保V的长度是(N+1)的倍数
if len(V) % (N + 1) == 0:
    # 按升序排序（按需）
    V.sort()

    # 计算所需子集数量
    num_subsets = len(V) // (N + 1)

    for i in range(num_subsets):
        subset = V[i * (N + 1): (i + 1) * (N + 1)]
        print(f"Subset {i + 1}:", subset)

        # 为子集内每个元素分配指定格式的索引
        indices_subset = [(j * 2 - 3 + i * (N + 1), -1 - i * (N + 1)) for j in range(1, N + 2)]
        print(f"Indices for Subset {i + 1}:", indices_subset)

else:
    print(f"V的长度({len(V)})不是{N+1}的倍数，无法分割为子集。")

当前输出 #

Subset 1: [3, 4, 5, 6]
Indices for Subset 1: [(-1, -1), (1, -1), (3, -1), (5, -1)]
Subset 2: [10, 11, 12, 13]
Indices for Subset 2: [(3, -5), (5, -5), (7, -5), (9, -5)]
Subset 3: [17, 18, 19, 20]
Indices for Subset 3: [(7, -9), (9, -9), (11, -9), (13, -9)]

期望输出 #

Subset 1: [3, 4, 5, 6]
Indices for Subset 1: [(-1, -1), (1, -1), (3, -1), (5, -1)]
Subset 2: [10, 11, 12, 13]
Indices for Subset 2: [(-1, -3), (1, -3), (3, -3), (5, -3)]
Subset 3: [17, 18, 19, 20]
Indices for Subset 3: [(-1, -5), (1, -5), (3, -5), (5, -5)]

问题分析与修复 #

对比当前与期望输出，索引生成公式存在两处错误：

1. 每个子集的第一个索引序列应固定为[-1, 1, 3, 5]，不需要叠加i*(N+1)的偏移量
2. 第二个索引的规律是-1, -3, -5，对应公式应为-1 - 2*i（i从0开始计数）

修正后的索引生成代码行如下：

indices_subset = [(j * 2 - 3, -1 - 2 * i) for j in range(1, N + 2)]

替换原代码中的对应行后，运行结果将与期望输出完全一致。

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