Méthode pythonique pour combiner deux listes de manière alternée ?

Question

J'ai deux listes, dont le premier est assuré de contenir exactement un élément de plus que le second . J'aimerais connaître la manière la plus pythonique de créer une nouvelle liste dont les valeurs d'indice pair proviennent de la première liste et dont les valeurs d'indice impair proviennent de la deuxième liste.

# example inputs
list1 = ['f', 'o', 'o']
list2 = ['hello', 'world']

# desired output
['f', 'hello', 'o', 'world', 'o']

Cela fonctionne, mais ce n'est pas joli :

list3 = []
while True:
    try:
        list3.append(list1.pop(0))
        list3.append(list2.pop(0))
    except IndexError:
        break

Comment y parvenir autrement ? Quelle est l'approche la plus pythique ?

Answer 1

Voici une façon de le faire en tranchant :

>>> 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']

Answer 2

Il y a une recette pour cela dans le itertools documentation :

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))

EDIT :

Pour les versions de python supérieures à 3 :

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))

Answer 3

En Python 2, cela devrait faire ce que vous voulez :

>>> iters = [iter(list1), iter(list2)]
>>> print list(it.next() for it in itertools.cycle(iters))
['f', 'hello', 'o', 'world', 'o']

Answer 4

Sans itertools et en supposant que l1 est 1 élément plus long que l2 :

>>> sum(zip(l1, l2+[0]), ())[:-1]
('f', 'hello', 'o', 'world', 'o')

Dans python 2, en utilisant itertools et en supposant que les listes ne contiennent pas None :

>>> filter(None, sum(itertools.izip_longest(l1, l2), ()))
('f', 'hello', 'o', 'world', 'o')

Answer 5

Je sais que la question porte sur deux listes dont l'une contient un élément de plus que l'autre, mais je me suis dit que je mettrais cela à la disposition des autres qui pourraient trouver cette question.

Voici La solution de Duncan adapté pour travailler avec deux listes de tailles différentes.

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

Ces sorties :

['f', 'hello', 'o', 'world', 'o', 'b', 'a', 'r']