纯列表实现：如何调整索引构造自定义起始位置的多维列表

需求：编写Python函数，接收shape参数（定义二维列表的维度）与若干values值，将values转换为指定维度的二维列表。当前存在索引变量v的问题：希望通过v控制二维列表每行的取值起始位置，例如示例中希望第一行从values的第4位（值为5）开始而非第0位（值为1），但因v初始值设为0无法实现该偏移。要求仅使用Python列表，禁止使用Numpy。

原代码：

def test(shape , *values):
    n = shape[1]
    m = shape[0]

    nDim = [[0 for column in range(m)] for row in range(n)]

    values = list(values)

    for x in range (n):
        print(x)
    
    
        v = 0
        if(v%m == 0):
            v +=m
    
        for y in range (m): 
            nDim[x][y] = values[v]
            v += 1

    return nDim

m = test((4,2) , 1, 2 ,3, 4 , 5 ,6 , 7 , 8)  

解决方案 #

核心问题是原代码中每次循环行时都会重置v为0，导致无法维持起始偏移。以下是两种可行的修改方案：

方案1：动态传入起始偏移量 #

给函数新增start_offset参数，支持灵活指定起始位置，同时保留默认逻辑兼容：

def test(shape, start_offset=0, *values):
    # 解析维度：shape[0]是每行元素数，shape[1]是行数
    row_len, row_count = shape
    nDim = [[0 for _ in range(row_len)] for _ in range(row_count)]
    values = list(values)
    
    # 校验偏移量合法性，避免索引越界
    if start_offset < 0 or start_offset + row_len * row_count > len(values):
        raise ValueError("起始偏移量超出values的有效范围")
    
    v = start_offset
    for x in range(row_count):
        for y in range(row_len):
            nDim[x][y] = values[v]
            v += 1
    return nDim

# 示例：从values的第4位（索引4，值为5）开始取值
result = test((4,2), 4, 1, 2, 3, 4, 5, 6, 7, 8)
print(result)
# 输出：[[5, 6, 7, 8], [1, 2, 3, 4]]

方案2：固定起始偏移量 #

如果不需要动态调整偏移，直接将v的初始值设为目标偏移量即可：

def test(shape , *values):
    row_len, row_count = shape
    nDim = [[0 for _ in range(row_len)] for _ in range(row_count)]
    values = list(values)
    
    # 固定从索引4开始取值
    v = 4
    for x in range(row_count):
        for y in range(row_len):
            nDim[x][y] = values[v]
            v += 1
            # 若需要循环复用values（超出长度后回到开头），可添加以下逻辑：
            # if v >= len(values):
            #     v = 0
    return nDim

result = test((4,2) , 1, 2 ,3, 4 , 5 ,6 , 7 , 8)
print(result)
# 输出：[[5, 6, 7, 8], [1, 2, 3, 4]]

关键修改说明 #

• 将v的初始化移到外层循环外，避免每次循环行时重置偏移
• 修正了shape的维度解析逻辑，直接解包更清晰
• 移除了原代码中无效的if(v%m == 0): v +=m逻辑，该逻辑会导致初始索引直接跳转到m，不符合需求

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