我正在使用itertools.chain以这种方式“拉平”列表列表:
itertools.chain
uniqueCrossTabs = list(itertools.chain(*uniqueCrossTabs))
这跟说的有什么不同?
uniqueCrossTabs = list(itertools.chain(uniqueCrossTabs))
所以如果uniqueCrossTabs是[ [ 1, 2 ], [ 3, 4 ] ],那就itertools.chain(*uniqueCrossTabs)等于说itertools.chain([ 1, 2 ], [ 3, 4 ])
uniqueCrossTabs是[ [ 1, 2 ], [ 3, 4 ] ]
itertools.chain(*uniqueCrossTabs)等于说itertools.chain([ 1, 2 ], [ 3, 4 ])
这与传递just显然不同uniqueCrossTabs。对于你的情况,你有一个要拼合的列表列表;什么itertools.chain()确实是在所有你传递给它的位置参数,其中每个位置参数是在自己的权利迭代拼接返回一个迭代。
uniqueCrossTabs
itertools.chain()
换句话说,你希望将每个列表uniqueCrossTabs作为参数传递给chain(),这会将它们链接在一起,但是你没有在单独的变量中使用列表,因此你可以使用*运算符将列表列表扩展为多个列表参数。
chain()
*
正如Jochen Ritzel在评论中指出的那样,chain.from_iterable()它更适合于此操作,因为它假定一个可迭代的对象开始。然后,你的代码将变得简单:
Jochen Ritzel
chain.from_iterable()
uniqueCrossTabs = list(itertools.chain.from_iterable(uniqueCrossTabs))