Как отсортировать (список / кортеж) списки / кортежи по элементу по заданному индексу? - PullRequest
535 голосов
/ 26 июня 2010

У меня есть некоторые данные в списке списков или в списке кортежей, например:

data = [[1,2,3], [4,5,6], [7,8,9]]
data = [(1,2,3), (4,5,6), (7,8,9)]

И я хочу отсортировать по второму элементу в подмножестве.Значение, сортировка по 2,5,8, где 2 от (1,2,3), 5 от (4,5,6).Каков общий способ сделать это?Должен ли я хранить кортежи или списки в моем списке?

Ответы [ 10 ]

939 голосов
/ 26 июня 2010
sorted_by_second = sorted(data, key=lambda tup: tup[1])

или

data.sort(key=lambda tup: tup[1])  # sorts in place
187 голосов
/ 11 ноября 2013
from operator import itemgetter
data.sort(key=itemgetter(1))
43 голосов
/ 18 ноября 2014

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

reverse = True

и результат будет следующим:

data.sort(key=lambda tup: tup[1], reverse=True)
32 голосов
/ 20 декабря 2015

Для сортировки по нескольким критериям, а именно, например, по второму и третьему элементам в кортеже, позвольте

data = [(1,2,3),(1,2,1),(1,1,4)]

и таким образом определите лямбду, которая возвращает кортеж, который описывает приоритет, например

sorted(data, key=lambda tup: (tup[1],tup[2]) )
[(1, 1, 4), (1, 2, 1), (1, 2, 3)]
27 голосов
/ 26 июня 2010

Ответ Стивена - тот, который я бы использовал.Для полноты, вот шаблон DSU (decorate-sort-undecorate) со списком:Однако , это было ненужным со времен Python 2.4, когда стали доступны ключевые функции.

13 голосов
/ 15 февраля 2017

Для сортировки списка кортежей (<word>, <count>), для count в порядке убывания и word в алфавитном порядке:

data = [
('betty', 1),
('bought', 1),
('a', 1),
('bit', 1),
('of', 1),
('butter', 2),
('but', 1),
('the', 1),
('was', 1),
('bitter', 1)]

Я использую этот метод:

sorted(data, key=lambda tup:(-tup[1], tup[0]))

и это дает мне результат:

[('butter', 2),
('a', 1),
('betty', 1),
('bit', 1),
('bitter', 1),
('bought', 1),
('but', 1),
('of', 1),
('the', 1),
('was', 1)]
9 голосов
/ 11 июля 2016

без лямбды:

def sec_elem(s):
    return s[1] 
sorted(data, key=sec_elem) 
6 голосов
/ 25 мая 2017

@ Ответ Стивена к сути!Вот пример для лучшей визуализации:

Привет фанатам Ready Player One! =)

>>> gunters = [('2044-04-05', 'parzival'), ('2044-04-07', 'aech'), ('2044-04-06', 'art3mis')]
>>> gunters.sort(key=lambda tup: tup[0])
>>> print gunters
[('2044-04-05', 'parzival'), ('2044-04-06', 'art3mis'), ('2044-04-07', 'aech')]

key - это функция, которая будет вызванапреобразовать элементы коллекции для сравнения .. как метод compareTo в Java.

Параметр, передаваемый ключу, должен вызываться.Здесь использование lambda создает анонимную функцию (которая вызывается).
Синтаксис lambda - это слово lambda, за которым следует повторяемое имя, а затем один блок кода.

Ниже приведен пример сортировки списка кортежей, который содержит информацию о времени определенного события и имени актера.

Мы сортируем этот список по времени возникновения события, которое является 0-м элементом кортежа.

Примечание - s.sort([cmp[, key[, reverse]]]) сортирует элементы s на месте

5 голосов
/ 27 сентября 2016

itemgetter() несколько быстрее, чем lambda tup: tup[1], но увеличение относительно невелико (около 10-25 процентов).

(сеанс IPython)

>>> from operator import itemgetter
>>> from numpy.random import randint
>>> values = randint(0, 9, 30000).reshape((10000,3))
>>> tpls = [tuple(values[i,:]) for i in range(len(values))]

>>> tpls[:5]    # display sample from list
[(1, 0, 0), 
 (8, 5, 5), 
 (5, 4, 0), 
 (5, 7, 7), 
 (4, 2, 1)]

>>> sorted(tpls[:5], key=itemgetter(1))    # example sort
[(1, 0, 0), 
 (4, 2, 1), 
 (5, 4, 0), 
 (8, 5, 5), 
 (5, 7, 7)]

>>> %timeit sorted(tpls, key=itemgetter(1))
100 loops, best of 3: 4.89 ms per loop

>>> %timeit sorted(tpls, key=lambda tup: tup[1])
100 loops, best of 3: 6.39 ms per loop

>>> %timeit sorted(tpls, key=(itemgetter(1,0)))
100 loops, best of 3: 16.1 ms per loop

>>> %timeit sorted(tpls, key=lambda tup: (tup[1], tup[0]))
100 loops, best of 3: 17.1 ms per loop
0 голосов
/ 04 марта 2014

Сортировка кортежа довольно проста:

tuple(sorted(t))
Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...