5

我的清单是:

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

我想将 mylist 转换为对列表:

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

有没有这样做的pythonic方式?列表理解?迭代工具?

4

5 回答 5

9

是的,列表理解是我通常的做法:

>>> groupsize = 2
>>> [mylist[x:x+groupsize] for x in range(0,len(mylist),groupsize)]
[[1,2],[3,4],[5,6]]
>>> groupsize = 3
>>> [mylist[x:x+groupsize] for x in range(0,len(mylist),groupsize)]
[[1,2,3],[4,5,6]]

range用于可移植性,如果您使用的是 python 2(您可能是)更改rangexrange以节省内存。

于 2012-08-03T02:55:05.607 回答
8

我的首选技术:

>>> mylist = [1, 2, 3, 4, 5, 6]
>>> mylist = iter(mylist)
>>> zip(mylist, mylist)
[(1, 2), (3, 4), (5, 6)]

无论如何,我通常使用生成器而不是列表,因此通常不需要第 2 行。

于 2012-08-03T03:06:22.347 回答
6

另一种方法:

zip( mylist[:-1:2], mylist[1::2] )

它产生一个元组列表:

>>> zip(mylist[:-1:2],mylist[1::2])
[(1, 2), (3, 4), (5, 6)]

如果你真的想要一个列表列表:

map(list, zip(mylist[:-1:2],mylist[1::2]))
于 2012-08-03T02:59:29.280 回答
5

查看itertools 文档中的“grouper”配方:

def grouper(n, iterable, fillvalue=None):
    "Collect data into fixed-length chunks or blocks"
    # grouper(3, 'ABCDEFG', 'x') --> ABC DEF Gxx
    args = [iter(iterable)] * n
    return izip_longest(fillvalue=fillvalue, *args)
于 2012-08-03T03:03:32.953 回答
2

[mylist[2*n:2*n+2] for n in xrange(len(mylist)/2)]

该解决方案结合使用列表推导和切片从原始列表中按顺序提取对,并构建切片列表。

或者,除了按两个而不是切片计数之外,[mylist[n:n+2] for n in xrange(0, len(mylist), 2)]这是相同的。xrange感谢 Steven Rumbalski 的建议。

而现在完全不同的东西:这是一个解决方案(ab)使用zip和一个临时函数而不是中间分配:

>>> (lambda i: zip(i, i))(iter(mylist))
[(1, 2), (3, 4), (5, 6)]
于 2012-08-03T02:52:28.657 回答