Tengo una lista de objetos:

array = [object0,object1,object2,object3,object4]

Y quiero cambiar el orden de los elementos dados una permutación:

permutation = [ 2 , 4 , 0 , 1 , 3 ]

¿Hay un comando en Python que haga algo como:

result = Permute(array,permutation)

result = [object2,object4,object0,object1,object3]

Sé que puedo hacerlo con un simple for bucle ...

4
Yochai Timmer 23 jun. 2011 a las 22:39

5 respuestas

La mejor respuesta

Si suponemos que permutation es una permutación adecuada de 0-n (cada una aparece exactamente una vez), entonces el siguiente código debería funcionar:

result=[array[i] for i in permutation]
5
murgatroid99 23 jun. 2011 a las 18:42

Utilice el método aleatorio de numpy

import numpy as np
arr = np.arange(10)
np.random.shuffle(arr)
print(arr)

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

Referencia: https: //docs.scipy. org / doc / numpy-1.15.0 / reference / generate / numpy.random.shuffle.html

0
ayaio 11 feb. 2019 a las 10:01

Solo por completo, una versión sin for en absoluto:

seed = ['foo', 'bar', 'baz']
permutation = [1, 2, 0]
result = map(lambda i: seed[i], permutation)
print result # --> ['bar', 'baz', 'foo']

Sin embargo, prefiero seguir con la lista de comprensión chicos. ;)

3
jena 23 jun. 2011 a las 18:47

En Python, esto es fácil de hacer con una comprensión de la lista:

result = [array[i] for i in permutation]
4
John Bartholomew 23 jun. 2011 a las 18:42

Puede usar el intercambio de índices. Tienes dos conjuntos a y b

def swap_random(a, b):
"""Randomly swap entries in two arrays."""
# Indices to swap
    swap_inds = np.random.random(size=len(a)) < 0.5 # your threshold 

# Make copies of arrays a and b for output
    a_out = np.copy(a)
    b_out = np.copy(b)

# Swap values
   a_out[swap_inds] = b[swap_inds]
   b_out[swap_inds] = a[swap_inds]

   return a_out, b_out

Entonces, haz la prueba

d = np.array(range(0,15))
r = np.array(range(16,31))

display(d,r)

>>> array([ 0,  1,  2,  3,  4,  5,  6,  7,  8,  9, 10, 11, 12, 13, 14])
>>> array([16, 17, 18, 19, 20, 21, 22, 23, 24, 25, 26, 27, 28, 29, 30])


display(swap_random(d, r))
>>> (array([ 0, 17,  2,  3, 20, 21, 22,  7, 24, 25, 10, 11, 28, 13, 14]),
>>> array([16,  1, 18, 19,  4,  5,  6, 23,  8,  9, 26, 27, 12, 29, 30]))
0
Vladimir Gavrysh 17 ago. 2019 a las 07:32