web-dev-qa-db-fra.com

Le moyen le plus rapide d'échanger des éléments dans la liste Python

Existe-t-il un moyen plus rapide d'échanger deux éléments de liste dans Python que

L[a], L[b] = L[b], L[a]

ou devrais-je avoir recours à Cython ou Weave ou similaire?

pythonperformance
46
Gerald Senarclens de Grancy

On dirait que le compilateur Python optimise le tuple temporaire avec cette construction:

code:

import dis

def swap1():
  a=5
  b=4
  a, b = b, a

def swap2():
  a=5
  b=4
  c = a
  a = b
  b = c

print 'swap1():'
dis.dis(swap1)
print 'swap2():'
dis.dis(swap2)

sortie:

swap1():
  6           0 LOAD_CONST               1 (5)
              3 STORE_FAST               0 (a)

  7           6 LOAD_CONST               2 (4)
              9 STORE_FAST               1 (b)

  8          12 LOAD_FAST                1 (b)
             15 LOAD_FAST                0 (a)
             18 ROT_TWO             
             19 STORE_FAST               0 (a)
             22 STORE_FAST               1 (b)
             25 LOAD_CONST               0 (None)
             28 RETURN_VALUE        
swap2():
 11           0 LOAD_CONST               1 (5)
              3 STORE_FAST               0 (a)

 12           6 LOAD_CONST               2 (4)
              9 STORE_FAST               1 (b)

 13          12 LOAD_FAST                0 (a)
             15 STORE_FAST               2 (c)

 14          18 LOAD_FAST                1 (b)
             21 STORE_FAST               0 (a)

 15          24 LOAD_FAST                2 (c)
             27 STORE_FAST               1 (b)
             30 LOAD_CONST               0 (None)
             33 RETURN_VALUE

Deux charges, un ROT_TWO , et deux sauvegardes, contre trois charges et trois sauvegardes. Il est peu probable que vous trouviez un mécanisme plus rapide.

126
Ignacio Vazquez-Abrams

Si vous pouviez publier un exemple de code représentatif, nous pourrions mieux analyser vos options. FWIW, pour la référence stupide suivante, j'obtiens environ une accélération 3x avec Shed Skin et une accélération 10x avec PyPy .

from time import time

def swap(L):
    for i in xrange(1000000):
        for b, a in enumerate(L):
            L[a], L[b] = L[b], L[a]

def main():
    start = time()
    L = list(reversed(range(100)))
    swap(L[:])
    print time() - start
    return L

if __== "__main__":
    print len(main())

# for shedskin:
# shedskin -b -r -e listswap.py && make
# python -c "import listswap; print len(listswap.main())"
3
TryPyPy