python - 从给定长度和宽度的一维列表创建二维列表？

Question

这个问题已经有答案了:

Create a 2D list out of 1D list (3 个回答)

已关闭 1 年前。

如何使用给定的行和列从一维列表创建二维列表？

给定:

sample_list=[5, 2, 3, 4, 1, 6, 1, 6, 7, 2, 3]

row=2
cols=4

它应该返回这个:

[[5, 2, 3, 4],[1, 6, 1, 6]]

我不需要其他数字 = 7, 2, 3。我只想要一个包含用户提供的行和列的列表。我的解决方案没有返回我想要的，

我的解决方案:

def 1D_to_2D(sample_list, row, cols):
    return [sample_list[i:i+cols] for i in range(0, len(sample_list), rows)]

返回:

[[5, 2, 3, 4], [3, 4, 1, 6], [1, 6, 1, 6], [1, 6, 7, 2], [7, 2, 3], [3]]

有人可以帮忙吗？

Answer 1

只需使用带有 range 的列表理解和 cols 步长(而不是您使用的 rows)来对列表进行切片，并限制使用 rows 进行外部切片的项目数:

sample_list=[5, 2, 3, 4, 1, 6, 1, 6, 7, 2, 3]
rows=2
cols=4

result = [sample_list[x:x+cols] for x in range(0,len(sample_list),cols)][:rows]

结果:

[[5, 2, 3, 4], [1, 6, 1, 6]]