229 votes

Liste tri basé sur des valeurs à partir d'une autre liste?

Je suis une liste de chaînes de caractères comme ceci:

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

Quel est le plus court chemin de la tri X en utilisant les valeurs de Y pour obtenir la sortie suivante?

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

L'ordre pour les éléments ayant le même "clé" n'a pas d'importance. Je peux recours à l'aide pour les constructions, mais je suis curieux de savoir si il existe un moyen plus court. Toutes les suggestions?

304voto

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

89voto

Ned Batchelder Points 128913

Zip les deux listes ensemble, à les trier, puis prendre la pièces que vous voulez:

>>> 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']

Les combiner ensemble pour obtenir:

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

53voto

Tom Points 341

Aussi, si vous n'avez pas l'esprit à l'aide de tableaux numpy (ou en fait déjà affaire avec numpy tableaux...), voici une autre solution sympa:

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]

Je l'ai trouvé ici: http://scienceoss.com/sort-one-list-by-another-list/

28voto

senderle Points 41607

La solution la plus évidente pour moi, c'est d'utiliser l' key mot-clé arg.

>>> 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']

Notez que vous pouvez la réduire à un one-liner si vous vous souciez de:

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

7voto

TMC Points 71

Une autre alternative, la combinaison de plusieurs réponses.

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

Prograide.com

Prograide est une communauté de développeurs qui cherche à élargir la connaissance de la programmation au-delà de l'anglais.
Pour cela nous avons les plus grands doutes résolus en français et vous pouvez aussi poser vos propres questions ou résoudre celles des autres.

Powered by: