25

可能重复:
如何在 Python 中将列表拆分为大小均匀的块?
python:将“5,4,2,4,1,0”转换为[[5, 4], [2, 4], [1, 0]]

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

->

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

有没有简单的方法来做到这一点,没有明确的'for'?

4

5 回答 5

67
>>> x = [1,2,3,4,5,6,7,8,9]
>>> zip(*[iter(x)]*3)
[(1, 2, 3), (4, 5, 6), (7, 8, 9)]

zip(*[iter(s)]*n)在 Python 中如何工作?

于 2012-04-28T14:28:52.050 回答
15

如果您真的希望子元素是列表与元组:

In [9]: [list(t) for t in zip(*[iter(range(1,10))]*3)]
Out[9]: [[1, 2, 3], [4, 5, 6], [7, 8, 9]]

或者,如果您想包含将被 截断的剩余元素zip,请使用切片语法:

In [16]: l=range(14)

In [17]: [l[i:i+3] for i in range(0,len(l),3)]
Out[17]: [[0, 1, 2], [3, 4, 5], [6, 7, 8], [9, 10, 11], [12, 13]]
于 2012-04-28T17:09:23.837 回答
10

您也可以numpy.reshape在这里使用:

import numpy as np

x = np.array([1,2,3,4,5,6,7,8,9])

new_x = np.reshape(x, (3,3))

结果:

>>> new_x
array([[1, 2, 3],
       [4, 5, 6],
       [7, 8, 9]])
于 2012-04-28T15:05:53.330 回答
8
>>> map(None,*[iter(s)]*3)
[(1, 2, 3), (4, 5, 6), (7, 8, 9)]
于 2012-04-28T15:15:51.077 回答
0

这是一种不那么“聪明”的递归方式:

from itertools import chain

def groupsof(n, xs):
    if len(xs) < n:
        return [xs]
    else:
        return chain([xs[0:n]], groupsof(n, xs[n:]))

print list(groupsof(3, [1,2,3,4,5,6,7,8,9,10,11,12,13]))
于 2012-04-28T15:08:06.283 回答