Question

我的列表看起来像这样的东西。我想知道是否有任何快速方法可以创建子列表？

list = [1,2,3,4,5,6,7,8,.....]

python将列表拆分到此子列表中的最佳方法是什么？

1. [(1, 2), (3, 4), (5, 6), ...]
2. [(1, 2, 3), (4, 5, 6), ...]
3. [(1,2,3,4),(5,6,7,8), ....]

Answer 1

您可以通过使用与每个子列表中所需元素数量相等的步骤（例如range(0, len(a), n)）循环遍历列表来轻松实现此目的。您可以使用索引（例如a[i:i+n]）提取每个子列表所需的元素。例如：

def group(a, n):
    return [tuple(a[i:i+n]) for i in range(0, len(a), n)]


print group([1, 2, 3, 4, 5, 6], 2)
# [(1, 2), (3, 4), (5, 6)]

print group([1, 2, 3, 4, 5, 6], 3)
# [(1, 2, 3), (4, 5, 6)]

print group([1, 2, 3, 4, 5, 6, 7, 8], 4)
# [(1, 2, 3, 4), (5, 6, 7, 8)]

Answer 2

尝试numpy.reshape：

>>> import numpy
>>> a = range(1,7)
>>> a
[1, 2, 3, 4, 5, 6]
>>> numpy.reshape(a,(3,2))
array([[1, 2],
       [3, 4],
       [5, 6]])
>>> numpy.reshape(a,(2,3))
array([[1, 2, 3],
       [4, 5, 6]])

reshape采用“数组类似”对象和包含新矩阵的行数和列数的元组。

拆分列表Python

2 个答案: