Une façon efficace de faire tourner une liste en python

Question

Quel est le moyen le plus efficace de faire tourner une liste en python ? Pour l'instant, j'ai quelque chose comme ça :

>>> def rotate(l, n):
...     return l[n:] + l[:n]
... 
>>> l = [1,2,3,4]
>>> rotate(l,1)
[2, 3, 4, 1]
>>> rotate(l,2)
[3, 4, 1, 2]
>>> rotate(l,0)
[1, 2, 3, 4]
>>> rotate(l,-1)
[4, 1, 2, 3]

Y a-t-il un meilleur moyen ?

Answer 1

A collections.deque est optimisé pour tirer et pousser aux deux extrémités. Ils ont même une rotate() méthode.

from collections import deque
items = deque([1, 2])
items.append(3)        # deque == [1, 2, 3]
items.rotate(1)        # The deque is now: [3, 1, 2]
items.rotate(-1)       # Returns deque to original state: [1, 2, 3]
item = items.popleft() # deque == [2, 3]

Answer 2

Et si vous utilisiez simplement pop(0) ?

list.pop([i])

Supprime l'élément à la position donnée dans la liste, et le renvoie. Si aucun indice n'est spécifié, a.pop() supprime et renvoie le dernier élément de la la liste. (Les crochets autour des i dans la signature de la méthode indiquent que le paramètre est optionnel, et non que vous devez taper des crochets à cet endroit. à cet endroit. Vous verrez fréquemment cette notation dans la référence de la bibliothèque Python).

Answer 3

Numpy peut le faire en utilisant la fonction roll commandement :

>>> import numpy
>>> a=numpy.arange(1,10) #Generate some data
>>> numpy.roll(a,1)
array([9, 1, 2, 3, 4, 5, 6, 7, 8])
>>> numpy.roll(a,-1)
array([2, 3, 4, 5, 6, 7, 8, 9, 1])
>>> numpy.roll(a,5)
array([5, 6, 7, 8, 9, 1, 2, 3, 4])
>>> numpy.roll(a,9)
array([1, 2, 3, 4, 5, 6, 7, 8, 9])

Answer 4

Cela dépend de ce que vous voulez qu'il se passe quand vous faites ça :

>>> shift([1,2,3], 14)

Vous pourriez vouloir changer votre :

def shift(seq, n):
    return seq[n:]+seq[:n]

à :

def shift(seq, n):
    n = n % len(seq)
    return seq[n:] + seq[:n]

Answer 5

Cela dépend également de la question de savoir si vous voulez déplacer la liste sur place (en la mutant), ou si vous voulez que la fonction renvoie une nouvelle liste. Parce que, d'après mes tests, quelque chose comme ça est au moins vingt fois plus rapide que votre implémentation qui ajoute deux listes :

def shiftInPlace(l, n):
    n = n % len(l)
    head = l[:n]
    l[:n] = []
    l.extend(head)
    return l

En fait, même en ajoutant un l = l[:] au sommet de cette liste pour opérer sur une copie de la liste transmise est toujours deux fois plus rapide.

Diverses implémentations avec un certain timing à http://gist.github.com/288272