Itérer sur toutes les paires d'éléments consécutifs dans une liste

Donné une liste

l = [1, 7, 3, 5]

Je veux effectuer une itération sur toutes les paires consécutives éléments de la liste de (1,7), (7,3), (3,5), c'est à dire

for i in xrange(len(l) - 1):
    x = l[i]
    y = l[i + 1]
    # do something

Je voudrais le faire de façon plus compacte, comme

for x, y in someiterator(l): ...

Est-il un moyen de faire cela à l'aide de Python itérateurs? Je suis sûr que le itertools module doit avoir une solution, mais je ne peux pas le comprendre.

  • Bien que je acceppted sberry réponse, comme je l'ai demandé un simple builtin à base de solution, pensez également à l'élégante et la plus performante des solutions par thefourtheye et HansZauber.
InformationsquelleAutor flonk | 2014-01-23
  1. 114

    Suffit d'utiliser zip 

    >>> l = [1, 7, 3, 5]
>>> for first, second in zip(l, l[1:]):
...     print first, second
...
1 7
7 3
3 5

    Comme suggéré que vous pourriez envisager d'utiliser la izip fonction dans itertools pour de très longues listes où vous ne voulez pas créer une nouvelle liste.

    import itertools

for first, second in itertools.izip(l, l[1:]):
    ...
    • Meh... en Python 2, zip() retourne une nouvelle liste. Mieux utiliser itertools.izip().
    • En Python 3, zip() retourne un itérateur. Mieux utiliser Python 3.
    • Merci, mais ne devrait-elle pas être zip(l[:-1], l[1:]) au lieu zip(l, l[1:])?
    • Cela crée une copie de l (la quasi-totalité de ses éléments) sans raison.
    • OK, votre commentaire ci-dessous les réponses que... 🙂
    • pas nécessairement en tant que zip vais essayer de faire des groupes complets.
    • c'est peut-être un problème, et c'est peut-être pas. L'OP l'exemple de l'entrée n'est certainement pas assez pour justifier de se soucier de telles choses. Si les performances grâce à un grand jeu de données est une préoccupation alors je pense que d'être une partie de la question.

    InformationsquelleAutor sberry
  2. 33

    Regarder pairwise à itertools recettes: http://docs.python.org/2/library/itertools.html#recipes

    Citant à partir de là:

    def pairwise(iterable):
    "s -> (s0,s1), (s1,s2), (s2, s3), ..."
    a, b = tee(iterable)
    next(b, None)
    return izip(a, b)

    Une Version Générale

    Une version générale, que les rendements des n-uplets d'un naturel positif de la taille, peut ressembler à ça:

    def nwise(iterable, n=2):                                                      
    iters = tee(iterable, n)                                                     
    for i, it in enumerate(iters):                                               
        next(islice(it, i, i), None)                                               
    return izip(*iters)
    • J'aime cette approche parce qu'elle n'est pas copier l'entrée itératif. Pour python3, il suffit d'utiliser zip au lieu de izip.
    • Comment étendre également inclure (sLast, s0)? Ainsi, au lieu de céder à n-1 paires, de retour de n paires?
    InformationsquelleAutor Bach
  3. 12

    Je voudrais créer un générique grouper générateur, à l'instar de ce

    def grouper(input_list, n = 2):
    for i in xrange(len(input_list) - (n - 1)):
        yield input_list[i:i+n]

    Exemple 1

    for first, second in grouper([1, 7, 3, 5, 6, 8], 2):
    print first, second

    Sortie

    1 7
7 3
3 5
5 6
6 8

    Exemple 1

    for first, second, third in grouper([1, 7, 3, 5, 6, 8], 3):
    print first, second, third

    Sortie

    1 7 3
7 3 5
3 5 6
5 6 8
    InformationsquelleAutor thefourtheye
  4. 0

    Vous pouvez utiliser un zip.

    >>> list(zip(range(5), range(2, 6)))
[(0, 1), (1, 2), (2, 3), (3, 4), (4, 5)]

    Comme une fermeture éclair, il crée des paires. Donc, pour mélanger vos deux listes, vous obtenez:

    >>> l = [1,7,3,5]
>>> list(zip(l[:-1], l[1:]))
[(1, 7), (7, 3), (3, 5)]

    Puis en itérant va comme

    for x, y in zip(l[:-1], l[1:]):
    pass
    • Vous n'avez pas besoin de couper la fin de la première zip ne fera que compléter les groupes. Ce serait différent si vous étiez à l'aide de izip_longest, mais alors, pourquoi voudriez-vous faire cela.
    • Vous avez raison, mais je l'aime mieux explicite, de cette façon. C'est quelque chose de personnel, je suppose.
    InformationsquelleAutor Noctua
  5. 0

    Généraliser sberry l'approche de la nwise avec la compréhension: l'

    def nwise(lst, k=2):
    return list(zip(*[lst[i:] for i in range(k)]))

    Par exemple

    nwise(list(range(10)),3)

    [(0, 1, 2), (1, 2, 3), (2, 3, 4), (3, 4, 5), (4, 5, 6), (5, 6,
    7), (6, 7, 8), (7, 8, 9)]

    InformationsquelleAutor alancalvitti
  6. -2

    Si vous voulais quelque chose de inline mais pas très lisible voici une autre solution qui rend l'utilisation de générateurs. Je m'attends c'est pas la meilleure performance sage :-/

    Convertir liste dans le générateur avec un réglage fin avant le dernier élément:

    gen = (x for x in l[:-1])

    Convertir en paires:

    [(gen.next(), x) for x in l[1:]]

    C'est tout ce dont vous avez besoin.

    • Pour l = [1, 2, 3, 4] ce produit [(1, 2), (3, 4)] et pas [(1, 2), (2, 3), (3, 4)] comme demandé. Il également ne fonctionne que lorsque la liste contient un nombre pair d'éléments.
    • Oups tu as raison. Je suis désolé, je ne devrais pas poster de la merde sur internet w/o de le tester. J'ai corrigé cela fonctionne maintenant (j'espère) si vous êtes intéressé par cette forme de solution, de toute façon.
    InformationsquelleAutor Burak Cetin