Làm cách nào để sắp xếp danh sách / bộ danh sách / bộ dữ liệu theo thành phần tại một chỉ mục nhất định?

Tôi có một số dữ liệu trong danh sách danh sách hoặc danh sách các bộ dữ liệu, như thế này:

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

Và tôi muốn sắp xếp theo phần tử thứ 2 trong tập hợp con. Có nghĩa là, sắp xếp theo 2,5,8 từ đâu 2đến (1,2,3), 5là từ (4,5,6). Cách phổ biến để làm điều này là gì? Tôi nên lưu trữ bộ dữ liệu hoặc danh sách trong danh sách của tôi?

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

hoặc là:

data.sort(key=lambda tup: tup[1])  # sorts in place

from operator import itemgetter
data.sort(key=itemgetter(1))

Tôi chỉ muốn thêm vào câu trả lời của Stephen nếu bạn muốn sắp xếp mảng từ cao xuống thấp, một cách khác ngoài các ý kiến ​​trên chỉ là thêm điều này vào dòng:

reverse = True

và kết quả sẽ như sau:

data.sort(key=lambda tup: tup[1], reverse=True)

Để sắp xếp theo nhiều tiêu chí, cụ thể là các phần tử thứ hai và thứ ba trong một tuple, hãy để

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

và vì vậy, xác định một lambda trả về một tuple mô tả mức độ ưu tiên, ví dụ

sorted(data, key=lambda tup: (tup[1],tup[2]) )
[(1, 1, 4), (1, 2, 1), (1, 2, 3)]

Câu trả lời của Stephen là câu tôi muốn sử dụng. Để hoàn thiện, đây là mẫu DSU (trang trí sắp xếp-không trang trí) với danh sách hiểu:

decorated = [(tup[1], tup) for tup in data]
decorated.sort()
undecorated = [tup for second, tup in decorated]

Hoặc, chặt chẽ hơn:

[b for a,b in sorted((tup[1], tup) for tup in data)]

Như đã lưu ý trong Hướng dẫn sắp xếp Python , điều này không cần thiết kể từ Python 2.4, khi các chức năng chính trở nên khả dụng.

Để sắp xếp danh sách các bộ dữ liệu (<word>, <count>), counttheo thứ tự giảm dần và wordtheo thứ tự bảng chữ cái:

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

Tôi sử dụng phương pháp này:

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

và nó cho tôi kết quả:

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

Không có lambda:

def sec_elem(s):
    return s[1]

sorted(data, key=sec_elem)

itemgetter() có phần nhanh hơn lambda tup: tup[1] , nhưng mức tăng tương đối khiêm tốn (khoảng 10 đến 25 phần trăm).

(Phiên 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

Câu trả lời của @Stephen là đến điểm! Dưới đây là một ví dụ để hình dung tốt hơn,

Hét lên cho người hâm mộ 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 là một chức năng sẽ được gọi để chuyển đổi các mục của bộ sưu tập để so sánh .. như compareTo phương thức trong Java.

Tham số được truyền cho khóa phải là một cái gì đó có thể gọi được. Ở đây, việc sử dụnglambda tạo một hàm ẩn danh (có thể gọi được).
Cú pháp của lambda là từ lambda theo sau là một tên lặp sau đó là một khối mã duy nhất.

Dưới đây ví dụ, chúng tôi đang sắp xếp một danh sách các tuple chứa thời gian abt thông tin của một số sự kiện và tên diễn viên nhất định.

Chúng tôi đang sắp xếp danh sách này theo thời gian xảy ra sự kiện - đó là yếu tố thứ 0 của một bộ dữ liệu.

Lưu ý - s.sort([cmp[, key[, reverse]]]) sắp xếp các mục của s tại chỗ

Sắp xếp một tuple khá đơn giản:

tuple(sorted(t))