Pythonic способ объединить два списка поочередно?
У меня есть два списка, первый из которых гарантированно содержит ровно на один элемент больше, чем второй. Я хотел бы знать наиболее Pythonic способ создать новый список, чьи значения четного индекса поступают из первого списка, а чьи значения нечетного индекса поступают из второго списка.
# example inputs
list1 = ['f', 'o', 'o']
list2 = ['hello', 'world']
# desired output
['f', 'hello', 'o', 'world', 'o']
Это работает, но не красиво:
list3 = []
while True:
try:
list3.append(list1.pop(0))
list3.append(list2.pop(0))
except IndexError:
break
Как еще это может быть достигнуто? Какой самый питоновский подход?
26 ответов
Вот один из способов сделать это нарезкой:
>>> list1 = ['f', 'o', 'o']
>>> list2 = ['hello', 'world']
>>> result = [None]*(len(list1)+len(list2))
>>> result[::2] = list1
>>> result[1::2] = list2
>>> result
['f', 'hello', 'o', 'world', 'o']
Есть рецепт для этого в itertools документация:
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))
import itertools
print [x for x in itertools.chain.from_iterable(itertools.izip_longest(list1,list2)) if x]
Я думаю, что это самый питонический способ сделать это.
Это должно делать то, что вы хотите:
>>> iters = [iter(list1), iter(list2)]
>>> print list(it.next() for it in itertools.cycle(iters))
['f', 'hello', 'o', 'world', 'o']
Если оба списка имеют одинаковую длину, вы можете:
[x for y in zip(list1, list2) for x in y]
Поскольку в первом списке есть еще один элемент, вы можете добавить его постфактум:
[x for y in zip(list1, list2) for x in y] + [list1[-1]]
Без itertools и предполагая, что l1 на 1 единицу длиннее l2:
>>> sum(zip(l1, l2+[0]), ())[:-1]
('f', 'hello', 'o', 'world', 'o')
Используя itertools и предполагая, что списки не содержат None:
>>> filter(None, sum(itertools.izip_longest(l1, l2), ()))
('f', 'hello', 'o', 'world', 'o')
Я знаю, что вопросы касаются двух списков, в одном из которых один элемент больше другого, но я решил, что я бы поставил его для других, кто может найти этот вопрос.
Вот решение Дункана, адаптированное для работы с двумя списками разных размеров.
list1 = ['f', 'o', 'o', 'b', 'a', 'r']
list2 = ['hello', 'world']
num = min(len(list1), len(list2))
result = [None]*(num*2)
result[::2] = list1[:num]
result[1::2] = list2[:num]
result.extend(list1[num:])
result.extend(list2[num:])
result
Это выводит:
['f', 'hello', 'o', 'world', 'o', 'b', 'a', 'r']
Вот один лайнер, который делает это:
list3 = [ item for pair in zip(list1, list2 + [0]) for item in pair][:-1]
def combine(list1, list2):
lst = []
len1 = len(list1)
len2 = len(list2)
for index in range( max(len1, len2) ):
if index+1 <= len1:
lst += [list1[index]]
if index+1 <= len2:
lst += [list2[index]]
return lst
from itertools import chain
list(chain(*zip('abc', 'def'))) # Note: this only works for lists of equal length
['a', 'd', 'b', 'e', 'c', 'f']
Этот основан на вкладе Карлоса Валиенте, приведенном выше, с возможностью чередовать группы из нескольких элементов и убедиться, что все элементы присутствуют в выходных данных:
A=["a","b","c","d"]
B=[1,2,3,4,5,6,7,8,9,10,11,12,13,14,15,16]
def cyclemix(xs, ys, n=1):
for p in range(0,int((len(ys)+len(xs))/n)):
for g in range(0,min(len(ys),n)):
yield ys[0]
ys.append(ys.pop(0))
for g in range(0,min(len(xs),n)):
yield xs[0]
xs.append(xs.pop(0))
print [x for x in cyclemix(A, B, 3)]
Это будет чередовать списки A и B группами по 3 значения в каждой:
['a', 'b', 'c', 1, 2, 3, 'd', 'a', 'b', 4, 5, 6, 'c', 'd', 'a', 7, 8, 9, 'b', 'c', 'd', 10, 11, 12, 'a', 'b', 'c', 13, 14, 15]
Может быть немного поздно, купить еще один Python с одним вкладышем. Это работает, когда два списка имеют одинаковый или неравный размер. Одна вещь ничего не стоит, это изменит a и b. Если это проблема, вам нужно использовать другие решения.
a = ['f', 'o', 'o']
b = ['hello', 'world']
sum([[a.pop(0), b.pop(0)] for i in range(min(len(a), len(b)))],[])+a+b
['f', 'hello', 'o', 'world', 'o']
Вот одна строка, использующая списки, без других библиотек:
list3 = [sub[i] for i in range(len(list2)) for sub in [list1, list2]] + [list1[-1]]
Вот другой подход, если вы разрешите изменение вашего начального list1 побочным эффектом:
[list1.insert((i+1)*2-1, list2[i]) for i in range(len(list2))]
itertools.zip_longest возвращает итератор пар кортежей, в котором все отсутствующие элементы в одном списке заменяются на
fillvalue=None (прохождение
fillvalue=object позволяет использовать
Noneкак значение). Если вы сгладите эти пары, отфильтруйте
fillvalue в понимании списка это дает:
>>> from itertools import zip_longest
>>> def merge(a, b):
... return [
... x for y in zip_longest(a, b, fillvalue=object)
... for x in y if x is not object
... ]
...
>>> merge("abc", "defgh")
['a', 'd', 'b', 'e', 'c', 'f', 'g', 'h']
>>> merge([0, 1, 2], [4])
[0, 4, 1, 2]
>>> merge([0, 1, 2], [4, 5, 6, 7, 8])
[0, 4, 1, 5, 2, 6, 7, 8]
Обобщено на произвольные итерации:
>>> def merge(*its):
... return [
... x for y in zip_longest(*its, fillvalue=object)
... for x in y if x is not object
... ]
...
>>> merge("abc", "lmn1234", "xyz9", [None])
['a', 'l', 'x', None, 'b', 'm', 'y', 'c', 'n', 'z', '1', '9', '2', '3', '4']
>>> merge(*["abc", "x"]) # unpack an iterable
['a', 'x', 'b', 'c']
Наконец, вы можете захотеть вернуть генератор, а не понимание списка:
>>> def merge(*its):
... return (
... x for y in zip_longest(*its, fillvalue=object)
... for x in y if x is not object
... )
...
>>> merge([1], [], [2, 3, 4])
<generator object merge.<locals>.<genexpr> at 0x000001996B466740>
>>> next(merge([1], [], [2, 3, 4]))
1
>>> list(merge([1], [], [2, 3, 4]))
[1, 2, 3, 4]
Если у вас все в порядке с другими пакетами, вы можете попробовать more_itertools.roundrobin:
>>> list(roundrobin('ABC', 'D', 'EF'))
['A', 'D', 'E', 'B', 'F', 'C']
Мой дубль:
a = "hlowrd"
b = "el ol"
def func(xs, ys):
ys = iter(ys)
for x in xs:
yield x
yield ys.next()
print [x for x in func(a, b)]
Как насчет NumPy? Он также работает со строками:
import numpy as np
np.array([[a,b] for a,b in zip([1,2,3],[2,3,4,5,6])]).ravel()
Результат:
array([1, 2, 2, 3, 3, 4])
def alternate_elements(small_list, big_list):
mew = []
count = 0
for i in range(len(small_list)):
mew.append(small_list[i])
mew.append(big_list[i])
count +=1
return mew+big_list[count:]
if len(l2)>len(l1):
res = alternate_elements(l1,l2)
else:
res = alternate_elements(l2,l1)
print(res)
Здесь мы меняем списки в зависимости от размера и производительности, может ли кто-нибудь предложить лучшее решение с временной сложностью O (len (l1) + len(l2))
Мой подход выглядит следующим образом:
from itertools import chain, zip_longest
def intersperse(*iterators):
# A random object not occurring in the iterators
filler = object()
r = (x for x in chain.from_iterable(zip_longest(*iterators, fillvalue=filler)) if x is not filler)
return r
list1 = ['f', 'o', 'o']
list2 = ['hello', 'world']
print(list(intersperse(list1, list2)))
Он работает для произвольного количества итераторов и дает итератор, поэтому я применилlist()в строке печати.
Останавливается на самое короткое:
def interlace(*iters, next = next) -> collections.Iterable:
"""
interlace(i1, i2, ..., in) -> (
i1-0, i2-0, ..., in-0,
i1-1, i2-1, ..., in-1,
.
.
.
i1-n, i2-n, ..., in-n,
)
"""
return map(next, cycle([iter(x) for x in iters]))
Конечно, разрешение метода next/__next__ может быть быстрее.
используя цикл for, мы также можем легко добиться этого:
list1 = ['f', 'o', 'o']
list2 = ['hello', 'world']
list3 = []
for i in range(len(list1)):
#print(list3)
list3.append(list1[i])
if i < len(list2):
list3.append(list2[i])
print(list3)
выход :
['f', 'hello', 'o', 'world', 'o']
Кроме того, используя понимание списка , это можно уменьшить. Но для понимания этот цикл можно использовать.
Несколько однострочников вдохновлены ответами на другой вопрос:
import itertools
list(itertools.chain.from_iterable(itertools.izip_longest(list1, list2, fillvalue=object)))[:-1]
[i for l in itertools.izip_longest(list1, list2, fillvalue=object) for i in l if i is not object]
[item for sublist in map(None, list1, list2) for item in sublist][:-1]
Альтернатива функциональным и неизменным способом (Python 3):
from itertools import zip_longest
from functools import reduce
reduce(lambda lst, zipped: [*lst, *zipped] if zipped[1] != None else [*lst, zipped[0]], zip_longest(list1, list2),[])
Явно опоздал на вечеринку, но вот краткий пример для списков равной длины:
output = [e for sub in zip(list1,list2) for e in sub]
Он также обобщает произвольное количество списков равной длины:
output = [e for sub in zip(list1,list2,list3) for e in sub]
и т.п.
Я бы сделал простое:
chain.from_iterable( izip( list1, list2 ) )
Он придумает итератор без создания каких-либо дополнительных потребностей хранения.
Это неприятно, но работает независимо от размера списков:
list3 = [element for element in list(itertools.chain.from_iterable([val for val in itertools.izip_longest(list1, list2)])) if element != None]
Я слишком стар, чтобы разбираться со списком, поэтому:
import operator
list3 = reduce(operator.add, zip(list1, list2))