Lista de clasificación según los valores correspondientes de una lista paralela [duplicate]

Question 1

Tengo una lista de cadenas como esta:

X = ["a", "b", "c", "d", "e", "f", "g", "h", "i"]
Y = [ 0,   1,   1,   0,   1,   2,   2,   0,   1 ]

¿Cuál es la forma más corta de ordenar X usando valores de Y para obtener el siguiente resultado?

["a", "d", "h", "b", "c", "e", "i", "f", "g"]

El orden de los elementos que tienen la misma “clave” no importa. Puedo recurrir al uso de for construye pero tengo curiosidad si hay una forma más corta. ¿Alguna sugerencia?

Question 2

Código más corto

[x for _, x in sorted(zip(Y, X))]

Ejemplo:

X = ["a", "b", "c", "d", "e", "f", "g", "h", "i"]
Y = [ 0,   1,   1,    0,   1,   2,   2,   0,   1]

Z = [x for _,x in sorted(zip(Y,X))]
print(Z)  # ["a", "d", "h", "b", "c", "e", "i", "f", "g"]

Generalmente hablando

[x for _, x in sorted(zip(Y, X), key=lambda pair: pair[0])]

Explicado:

zip los dos lists.
crear un nuevo, ordenado list basado en el zip usando sorted().
usando una lista de comprensión extracto los primeros elementos de cada par de los ordenados, comprimidos list.

Para obtener más información sobre cómo configurar\usar el key parámetro, así como el sorted función en general, eche un vistazo a este.

Question 3

Comprima las dos listas, ordénelas y luego tome las partes que desee:

>>> yx = zip(Y, X)
>>> yx
[(0, 'a'), (1, 'b'), (1, 'c'), (0, 'd'), (1, 'e'), (2, 'f'), (2, 'g'), (0, 'h'), (1, 'i')]
>>> yx.sort()
>>> yx
[(0, 'a'), (0, 'd'), (0, 'h'), (1, 'b'), (1, 'c'), (1, 'e'), (1, 'i'), (2, 'f'), (2, 'g')]
>>> x_sorted = [x for y, x in yx]
>>> x_sorted
['a', 'd', 'h', 'b', 'c', 'e', 'i', 'f', 'g']

Combina estos para obtener:

[x for y, x in sorted(zip(Y, X))]

Question 4

Además, si no le importa usar matrices numpy (o, de hecho, ya está tratando con matrices numpy…), aquí hay otra buena solución:

people = ['Jim', 'Pam', 'Micheal', 'Dwight']
ages = [27, 25, 4, 9]

import numpy
people = numpy.array(people)
ages = numpy.array(ages)
inds = ages.argsort()
sortedPeople = people[inds]

Lo encontré aquí:
http://scienceoss.com/ordenar-una-lista-por-otra-lista/

Question 5

La solución más obvia para mí es usar el key palabra clave argumento

>>> X = ["a", "b", "c", "d", "e", "f", "g", "h", "i"]
>>> Y = [ 0,   1,   1,    0,   1,   2,   2,   0,   1]
>>> keydict = dict(zip(X, Y))
>>> X.sort(key=keydict.get)
>>> X
['a', 'd', 'h', 'b', 'c', 'e', 'i', 'f', 'g']

Tenga en cuenta que puede acortar esto a una sola línea si desea:

>>> X.sort(key=dict(zip(X, Y)).get)

Como han señalado Wenmin Mu y Jack Peng, esto supone que los valores en X son todos distintos. Eso se maneja fácilmente con una lista de índice:

>>> Z = ["A", "A", "C", "C", "C", "F", "G", "H", "I"]
>>> Z_index = list(range(len(Z)))
>>> Z_index.sort(key=keydict.get)
>>> Z = [Z[i] for i in Z_index]
>>> Z
['A', 'C', 'H', 'A', 'C', 'C', 'I', 'F', 'G']

Dado que el enfoque decorar-clasificar-desdecorar descrito por Whatang es un poco más simple y funciona en todos los casos, probablemente sea mejor la mayor parte del tiempo. (¡Esta es una respuesta muy antigua!)

Question 6

more_itertools tiene una herramienta para ordenar iterables en paralelo:

Dado

from more_itertools import sort_together


X = ["a", "b", "c", "d", "e", "f", "g", "h", "i"]
Y = [ 0,   1,   1,    0,   1,   2,   2,   0,   1]

Manifestación

sort_together([Y, X])[1]
# ('a', 'd', 'h', 'b', 'c', 'e', 'i', 'f', 'g')

Question 7

De hecho, vine aquí buscando ordenar una lista por una lista donde los valores coincidieran.

list_a = ['foo', 'bar', 'baz']
list_b = ['baz', 'bar', 'foo']
sorted(list_b, key=lambda x: list_a.index(x))
# ['foo', 'bar', 'baz']

Question 8

Otra alternativa, combinar varias de las respuestas.

zip(*sorted(zip(Y,X)))[1]

Para trabajar para python3:

list(zip(*sorted(zip(B,A))))[1]

¿Ha sido útil esta solución?