Вопрос:

Сортировка списка на основе значений из другого списка?

python sorting

207516 просмотра

14 ответа

56788 Репутация автора

У меня есть список строк, как это:

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

Какой самый короткий способ сортировки X, используя значения из Y, чтобы получить следующий вывод?

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

Порядок элементов, имеющих одинаковый «ключ», не имеет значения. Я могу прибегнуть к использованию forконструкций, но мне любопытно, если есть более короткий путь. Какие-либо предложения?

Автор: Legend Источник Размещён: 07.07.2011 11:56

Ответы (14)


354 плюса

4949 Репутация автора

Решение

Кратчайший код

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

Пример:

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

Вообще говоря

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

Разъяснение:

  1. zipдва listс.
  2. создать новый, отсортированный в listзависимости от zipиспользования sorted().
  3. используя понимание списка, извлеките первые элементы каждой пары из отсортированного, сжатого list.

Для получения дополнительной информации о том, как установить \ использовать keyпараметр, а также sortedфункцию в целом, взгляните на это .


Автор: Whatang Размещён: 08.07.2011 12:02

31 плюса

97130 Репутация автора

Самым очевидным решением для меня является использование keyключевого слова 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']

Обратите внимание, что вы можете сократить это до одной строки, если вы хотите:

>>> X.sort(key=dict(zip(X, Y)).get)
Автор: senderle Размещён: 08.07.2011 12:02

95 плюса

267756 Репутация автора

Сожмите два списка вместе, рассортируйте их, затем возьмите нужные части:

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

Объедините их вместе, чтобы получить:

[x for y, x in sorted(zip(Y, X))]
Автор: Ned Batchelder Размещён: 08.07.2011 12:03

4 плюса

7706 Репутация автора

zip, сортировка по второму столбцу, возврат первого столбца.

zip(*sorted(zip(X,Y), key=operator.itemgetter(1)))[0]
Автор: riza Размещён: 08.07.2011 05:07

10 плюса

171 Репутация автора

Еще одна альтернатива, объединяющая несколько ответов.

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

Чтобы работать на python3:

list(zip(*sorted(zip(B,A))))[1]
Автор: TMC Размещён: 15.10.2013 01:21

63 плюса

1244 Репутация автора

Кроме того, если вы не возражаете против использования numpy-массивов (или на самом деле уже имеете дело с numpy-массивами ...), вот еще одно приятное решение:

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]

Я нашел это здесь: http://scienceoss.com/sort-one-list-by-another-list/

Автор: Tom Размещён: 12.01.2014 04:18

12 плюса

161 Репутация автора

Мне нравится иметь список отсортированных индексов. Таким образом, я могу отсортировать любой список в том же порядке, что и список источников. Как только у вас есть список отсортированных индексов, простое понимание списка сделает свое дело:

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

    sorted_y_idx_list = sorted(range(len(Y)),key=lambda x:Y[x])
    Xs = [X[i] for i in sorted_y_idx_list ]

    print( "Xs:", Xs )
    # prints: Xs: ["a", "d", "h", "b", "c", "e", "i", "f", "g"]

Обратите внимание, что отсортированный список индексов также можно получить с помощью numpy argsort ().

Автор: 1-ijk Размещён: 15.03.2015 09:47

5 плюса

15356 Репутация автора

more_itertools имеет инструмент для параллельной сортировки итераций:

from more_itertools import sort_together


sort_together([Y, X])[1]
# ('a', 'd', 'h', 'b', 'c', 'e', 'i', 'f', 'g')
Автор: pylang Размещён: 04.08.2017 07:53

1 плюс

1527 Репутация автора

Вы можете создать pandas Series, используя первичный список как dataи другой список как index, а затем просто отсортировать по индексу:

import pandas as pd
pd.Series(data=X,index=Y).sort_index().tolist()

выход:

['a', 'd', 'h', 'b', 'c', 'e', 'i', 'f', 'g']
Автор: Binyamin Even Размещён: 07.01.2018 11:53

2 плюса

321 Репутация автора

Быстрый однострочник.

list_a = [5,4,3,2,1]
list_b = [1,1.5,1.75,2,3,3.5,3.75,4,5]

Скажем, вы хотите, чтобы список a соответствовал списку b.

orderedList =  sorted(list_a, key=lambda x: list_b.index(x))

Это полезно, когда нужно упорядочить меньший список со значениями в большем. Предполагая, что большой список содержит все значения в меньшем списке, это можно сделать.

Автор: Evan Lalo Размещён: 09.01.2018 08:49

0 плюса

9 Репутация автора

list1 = ['a','b','c','d','e','f','g','h','i']
list2 = [0,1,1,0,1,2,2,0,1]

output=[]
cur_loclist = []

Чтобы получить уникальные значения, присутствующие в list2

list_set = set(list2)

Найти местоположение индекса в list2

list_str = ''.join(str(s) for s in list2)

Расположение индекса в list2отслеживается с помощьюcur_loclist

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

for i in list_set:
cur_loc = list_str.find(str(i))

while cur_loc >= 0:
    cur_loclist.append(cur_loc)
    cur_loc = list_str.find(str(i),cur_loc+1)

print(cur_loclist)

for i in range(0,len(cur_loclist)):
output.append(list1[cur_loclist[i]])
print(output)
Автор: VANI Размещён: 16.02.2018 05:03

0 плюса

971 Репутация автора

Вот ответ Whatangs, если вы хотите получить оба отсортированных списков (python3).

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

Zx, Zy = zip(*[(x, y) for x, y in sorted(zip(Y, X))])

print(list(Zx))  # [0, 0, 0, 1, 1, 1, 1, 2, 2]
print(list(Zy))  # ['a', 'd', 'h', 'b', 'c', 'e', 'i', 'f', 'g']

Просто помните, что Zx и Zy - это кортежи. Я также брожу, если есть лучший способ сделать это.

Предупреждение: если вы запускаете его с пустыми списками, он вылетает.

Автор: Anoroah Размещён: 02.03.2018 03:57

1 плюс

933 Репутация автора

Я создал более общую функцию, которая сортирует более двух списков на основе другого, вдохновленного ответом @ Whatang.

def parallel_sort(*lists):
    """
    Sorts the given lists, based on the first one.
    :param lists: lists to be sorted

    :return: a tuple containing the sorted lists
    """

    # Create the initially empty lists to later store the sorted items
    sorted_lists = tuple([] for _ in range(len(lists)))

    # Unpack the lists, sort them, zip them and iterate over them
    for t in sorted(zip(*lists)):
        # list items are now sorted based on the first list
        for i, item in enumerate(t):    # for each item...
            sorted_lists[i].append(item)  # ...store it in the appropriate list

    return sorted_lists
Автор: pgmank Размещён: 26.03.2018 03:12

4 плюса

2296 Репутация автора

Я на самом деле пришел сюда, чтобы отсортировать список по списку, где значения совпадают.

list_a = ['foo', 'bar', 'baz']
list_b = ['baz', 'bar', 'foo']
sorted(list_b, key=lambda x: list_a.index(x))
# ['foo', 'bar', 'baz']
Автор: nackjicholson Размещён: 17.08.2018 11:07
Вопросы из категории :
32x32