Python: отсортировать список и последовательно изменить другой - PullRequest
16 голосов
/ 29 апреля 2010

У меня есть два списка: один содержит множество точек x, другой содержит y точек. Python каким-то образом удается смешать х точек вверх, или пользователь может. Мне нужно отсортировать их по убыванию и переместить точки y, чтобы они следовали за их x-соответствиями. Они в двух отдельных списках .. как мне это сделать?

Ответы [ 4 ]

18 голосов
/ 29 апреля 2010

Вы можете сжать списки и отсортировать результат. Сортировка кортежей должна по умолчанию сортироваться по первому элементу.

>>> xs = [3,2,1]
>>> ys = [1,2,3]
>>> points = zip(xs,ys)
>>> points
[(3, 1), (2, 2), (1, 3)]
>>> sorted(points)
[(1, 3), (2, 2), (3, 1)]

А затем снова распаковать их:

>>> sorted_points = sorted(points)
>>> new_xs = [point[0] for point in sorted_points]
>>> new_ys = [point[1] for point in sorted_points]
>>> new_xs
[1, 2, 3]
>>> new_ys
[3, 2, 1]
16 голосов
/ 29 апреля 2010
>>> xs = [5, 2, 1, 4, 6, 3]
>>> ys = [1, 2, 3, 4, 5, 6]
>>> xs, ys = zip(*sorted(zip(xs, ys)))
>>> xs
(1, 2, 3, 4, 5, 6)
>>> ys
(3, 2, 6, 4, 1, 5)
10 голосов
/ 29 апреля 2010
>>> import numpy

>>> sorted_index = numpy.argsort(xs)
>>> xs = [xs[i] for i in sorted_index]
>>> ys = [ys[i] for i in sorted_index]

если вы можете работать с numpy.array

>>> xs = numpy.array([3,2,1])
>>> xs = numpy.array([1,2,3])
>>> sorted_index = numpy.argsort(xs)
>>> xs = xs[sorted_index]
>>> ys = ys[sorted_index]
4 голосов
/ 29 апреля 2010

Если x и y означают одну единицу (например, точку), было бы более целесообразно хранить их как кортежи, а не как два отдельных списка.

Независимо от того, что вы должны сделать:

x = [4, 2, 5, 4, 5,…]
y = [4, 5, 2, 3, 1,…]

zipped_list = zip(x,y)
sorted_list = sorted(zipped_list)
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...