有效地将平面列表切成多层嵌套列表

Question

例如，我有一个简单的清单

 [1, 2, 3, 4, 5, 6, 7, 8, 9, 'A', 'B', 'C', 'D', 'E', 'F', 'G']

我想将其转换为4深列表

[[[[1, 2], [3, 4]], [[5, 6], [7, 8]]], [[[9, 'A'], ['B', 'C']], [['D', 'E'] ['F', 'G']]]]

有没有一种方法可以为每个级别创建一个单独的变量？ 什么是最节省内存和性能的方法？

更新：另外，有没有办法以非对称的方式做到这一点？

[[[[1, 2, 3], 4], [[5, 6, 7], 8]]], [[[9, 'A', 'B'], 'C']], [['D', 'E', 'F'], 'G']]]]

Answer 1

请注意，您的第一个列表包含15个元素而不是16个元素。此外， A应该是什么？ 您在其他地方定义的常数吗？ 我只是假设它是一个字符串： 'A' 。

如果你有工作np.arrays ，你可以简单地重塑你的数组：

import numpy as np
r = np.array([1, 2, 3, 4, 5, 6, 7, 8, 9, 'A', 'B', 'C', 'D', 'E', 'F', 'G'])
r.reshape(2,2,2,2)

它输出：

array([[[['1', '2'],
         ['3', '4']],

        [['5', '6'],
         ['7', '8']]]


       [[['9', 'A'],
         ['B', 'C']],

        [['D', 'E'],
         ['F', 'G']]]
      dtype='<U11')

这应该非常有效，因为numpy不会更改基础数据格式。 它仍然是平面阵列，显示方式不同。

Numpy不支持不规则形状。 然后，您必须使用标准的python列表：

i = iter([1, 2, 3, 4, 5, 6, 7, 8, 9, 'A', 'B', 'C', 'D', 'E', 'F', 'G'])

l1 = []

for _ in range(2):
    l2 = []
    for _ in range(2):
        l3 = []
        l4 = []
        for _ in range(3):
            l4.append(next(i))
        l3.append(l4)
        l3.append(next(i))
        l2.append(l3)
    l1.append(l2)

print(l1)
# [[[[1, 2, 3], 4], [[5, 6, 7], 8]], [[[9, 'A', 'B'], 'C'], [['D', 'E', 'F'], 'G']]]

如您所说，您必须为每个级别定义一个临时变量。 我想您可以使用列表推导，但是它们并不漂亮。