Python-将一维列表转换为可变长度的二维列表

python中的列表通常是一维列表，其中元素依次列出。但是在2D列表中，列表嵌套在外部列表中。在本文中，我们将看到如何从给定的1D列表中创建2D列表。我们还向程序提供2D列表内元素数量的值。

使用append和index

在这种方法中，我们将创建一个for循环，以循环浏览2D列表中的每个元素，并将其用作要创建的新列表的索引。我们通过从零开始并将其添加到从2D列表中接收到的元素中来不断增加索引值。

示例

# 给定列表
listA = ['Sun','Mon','Tue','Wed','Thu','Fri','Sat']

# 需要2D列表的长度
len_2d = [ 2, 4]

#声明空白的新列表
res = []
def convert(listA, len_2d):
   idx = 0
   for var_len in len_2d:
      res.append(listA[idx: idx + var_len])
      idx += var_len
convert(listA, len_2d)
print("The new 2D lis is: \n",res)

运行上面的代码给我们以下结果-

输出结果

The new 2D lis is:
[[1, 2], [3, 4, 5, 6]]

使用islice

islice函数可用于根据2D列表的要求对给定列表与一定数量的元素进行切片。因此，本周我们将浏览2D列表的每个元素，并使用该值2分割原始列表。我们需要itertools包才能使用islice函数。

示例

from itertools import islice
# 给定列表
listA = ['Sun','Mon','Tue','Wed','Thu','Fri','Sat']

# 需要2D列表的长度
len_2d = [ 3, 2]

# 使用islice
def convert(listA, len_2d):
   res = iter(listA)
   return [list(islice(res,i)) for i in len_2d]
res = [convert(listA, len_2d)]
print("The new 2D lis is: \n",res)

运行上面的代码给我们以下结果-

输出结果

The new 2D lis is:
[[['Sun', 'Mon', 'Tue'], ['Wed', 'Thu']]]