假设我有两个列表，一个比另一个长，x = [1,2,3,4,5,6,7,8]并且y = [a,b,c]我想将 y 中的每个元素合并到 x 中的每个第三个索引，因此结果列表 z 看起来像：z = [1,2,a,3,4,b,5,6,c,7,8]
在 python 中解决这个问题的最佳方法是什么？
问问题
1894 次
8 回答
5
这是来自itertools 文档的轮询配方的改编版本，应该可以满足您的要求：
from itertools import cycle, islice
def merge(a, b, pos):
"merge('ABCDEF', [1,2,3], 3) --> A B 1 C D 2 E F 3"
iterables = [iter(a)]*(pos-1) + [iter(b)]
pending = len(iterables)
nexts = cycle(iter(it).next for it in iterables)
while pending:
try:
for next in nexts:
yield next()
except StopIteration:
pending -= 1
nexts = cycle(islice(nexts, pending))
例子：
>>> list(merge(xrange(1, 9), 'abc', 3)) # note that this works for any iterable!
[1, 2, 'a', 3, 4, 'b', 5, 6, 'c', 7, 8]
或者，您可以roundrobin()在不进行任何修改的情况下按原样使用：
>>> x = [1,2,3,4,5,6,7,8]
>>> y = ['a','b','c']
>>> list(roundrobin(*([iter(x)]*2 + [y])))
[1, 2, 'a', 3, 4, 'b', 5, 6, 'c', 7, 8]
或等效但更具可读性的版本：
>>> xiter = iter(x)
>>> list(roundrobin(xiter, xiter, y))
[1, 2, 'a', 3, 4, 'b', 5, 6, 'c', 7, 8]
请注意，这两种方法都适用于任何可迭代的，而不仅仅是序列。
这是原始roundrobin()实现：
from itertools import cycle, islice
def roundrobin(*iterables):
"roundrobin('ABC', 'D', 'EF') --> A D E B F C"
# Recipe credited to George Sakkis
pending = len(iterables)
nexts = cycle(iter(it).next for it in iterables)
while pending:
try:
for next in nexts:
yield next()
except StopIteration:
pending -= 1
nexts = cycle(islice(nexts, pending))
于 2013-06-21T18:53:07.830 回答
3
这是另一种方式：
x = range(1, 9)
y = list('abc')
from itertools import count, izip
from operator import itemgetter
from heapq import merge
print map(itemgetter(1), merge(enumerate(x), izip(count(1, 2), y)))
# [1, 2, 'a', 3, 4, 'b', 5, 6, 'c', 7, 8]
这使它在构建新列表之前保持懒惰，并让merge自然地合并序列......有点装饰/不装饰......虽然它确实需要 Python 2.7count有一个step参数。
所以，稍微介绍一下：
a = list(enumerate(x))
# [(0, 1), (1, 2), (2, 3), (3, 4), (4, 5), (5, 6), (6, 7), (7, 8)]
b = zip(count(1, 2), y)
# [(1, 'a'), (3, 'b'), (5, 'c')]
print list(merge(a, b))
# [(0, 1), (1, 2), (1, 'a'), (2, 3), (3, 4), (3, 'b'), (4, 5), (5, 6), (5, 'c'), (6, 7), (7, 8)]
然后itemgetter(1)只取删除索引的实际值......
于 2013-06-21T19:38:06.023 回答
3
这种方法x就地修改。x或者，如果您不想更改原件，您可以复制并返回修改后的副本。
def merge(x, y, offset):
for i, element in enumerate(y, 1):
x.insert(i * offset - 1, element)
>>> x = [1,2,3,4,5,6,7,8]
>>> y = ['a','b','c']
>>> merge(x, y, 3)
>>> x
[1, 2, 'a', 3, 4, 'b', 5, 6, 'c', 7, 8]
y结束后的所有额外元素都x将附加到末尾。
于 2013-06-21T19:00:04.873 回答
3
>>> from itertools import chain
def solve(x,y):
it = iter(y)
for i in xrange(0, len(x), 2):
try:
yield x[i:i+2] + [next(it)]
except StopIteration:
yield x[i:]
...
>>> x = [1,2,3,4,5,6,7,8]
>>> y = ['a','b','c']
>>> list(chain.from_iterable(solve(x,y)))
[1, 2, 'a', 3, 4, 'b', 5, 6, 'c', 7, 8]
于 2013-06-21T18:46:52.283 回答
0
使用itertools.izip_longest：
>>> from itertools import izip_longest, chain
>>> y = ['a','b','c']
>>> x = [1,2,3,4,5,6,7,8]
>>> lis = (x[i:i+2] for i in xrange(0, len(x) ,2)) # generator expression
>>> list(chain.from_iterable([ (a + [b]) if b else a
for a, b in izip_longest(lis, y)]))
[1, 2, 'a', 3, 4, 'b', 5, 6, 'c', 7, 8]
于 2013-06-21T19:21:52.467 回答
0
sep, lst = 2, []
for i in range(len(y)+1):
lst += x[i*sep:(i+1)*sep] + y[i:i+1]
插入元素之前的元素sep数量在哪里。xy
表现：
>>> timeit.timeit(stmt="for i in range(len(y)+1): lst += x[i*sep:(i+1)*sep] + y[i:i+1]", setup="lst = [];x = [1,2,3,4,5,6,7,8];y = ['a','b','c'];sep = 2", number=1000000)
2.8565280437469482
相当不错。我无法stmt开始，let = []所以我认为它一直在附加lst（除非我误解了timeit），但仍然......一百万次相当不错。
于 2013-06-21T19:04:58.113 回答
0
上面的解决方案真的很酷。这是一个不涉及循环或迭代工具的替代方案。
def merge(x, y):
result = []
while y:
for i in range(0, 2): result.append(x.pop(0))
for i in range(0, 1): result.append(y.pop(0))
result.extend(x)
return result
其中 2 和 1 是任意的，并且假定列表 y 比列表 x 短。
于 2013-06-21T19:01:51.133 回答
0
def merge(xs, ys):
ys = iter(ys)
for i, x in enumerate(xs, 1):
yield x
if i % 2 == 0:
yield next(ys)
''.join(merge('12345678', 'abc')) # => '12a34b56c78'
于 2013-06-21T19:30:09.847 回答