Hướng dẫn python groupby list of objects - python nhóm theo danh sách các đối tượng

Gần đây tôi đã gặp phải một vấn đề mà tôi phải nhóm các đối tượng của một danh sách duy nhất theo giá trị của một thuộc tính của các đối tượng này. Sau khi thử một số điều, tôi đã giải quyết chức năng

get_attr = operator.attrgetter('some_attr')
new_list = [list(g) for k, g in itertools.groupby(sorted(old_list, key=get_attr), get_attr)]

get_attr = operator.attrgetter('some_attr')
new_list = [list(g) for k, g in itertools.groupby(sorted(old_list, key=get_attr), get_attr)]

Hãy xem xét ví dụ sau, chúng tôi có một lớp chữ cái với hai thuộc tính: một chuỗi để chỉ định ký tự và boolean để chỉ định chữ cái là nguyên âm hay không. Chúng tôi muốn nhóm các đối tượng thư dựa trên thuộc tính

get_attr = operator.attrgetter('some_attr')
new_list = [list(g) for k, g in itertools.groupby(sorted(old_list, key=get_attr), get_attr)]

Lớp này trông như sau,

get_attr = operator.attrgetter('some_attr')
new_list = [list(g) for k, g in itertools.groupby(sorted(old_list, key=get_attr), get_attr)]

class Letter:
    def __init__(self, char: str, vowel: bool) -> None:
        self.char = char
        self.vowel = vowel

    def __repr__(self) -> str:
        return self.char

Tiếp theo chúng tôi tạo một danh sách với các đối tượng chữ cái. Không phải toàn bộ bảng chữ cái, các chữ cái đầu tiên sẽ làm.

letters = [
    Letter('a', True),
    Letter('b', False),
    Letter('c', False),
    Letter('d', False),
    Letter('e', True),
]

Bây giờ chúng tôi nhóm danh sách

get_attr = operator.attrgetter('some_attr')
new_list = [list(g) for k, g in itertools.groupby(sorted(old_list, key=get_attr), get_attr)]

get_item = operator.itemgetter(some_index)
new_list = [list(g) for k, g in itertools.groupby(sorted(old_list, key=get_item), get_item)]

get_item = operator.itemgetter(some_index)
new_list = [list(g) for k, g in itertools.groupby(sorted(old_list, key=get_item), get_item)]

get_attr = operator.attrgetter('some_attr')
new_list = [list(g) for k, g in itertools.groupby(sorted(old_list, key=get_attr), get_attr)]

from itertools import groupby

sorted_letters = sorted(letters, key=lambda letter: letter.vowel)
grouped = [list(result) for key, result in groupby(
    sorted_letters, key=lambda letter: letter.vowel)]

print(grouped)
# => [[b, c, d], [a, e]]

Một gotcha là danh sách, hoặc bất kỳ điều gì khác, cần phải được sắp xếp trước khi chuyển nó đến hàm

get_attr = operator.attrgetter('some_attr')
new_list = [list(g) for k, g in itertools.groupby(sorted(old_list, key=get_attr), get_attr)]

get_attr = operator.attrgetter('some_attr')
new_list = [list(g) for k, g in itertools.groupby(sorted(old_list, key=get_attr), get_attr)]

from itertools import groupby

# letters = sorted(letters, key=lambda letter: letter.vowel)
grouped = [list(result) for key, result in groupby(
    letters, key=lambda letter: letter.vowel)]

print(grouped)
# => [[a], [b, c, d], [e]]

Toàn bộ mẫu mã của chúng tôi trông giống như:

from itertools import groupby


class Letter:
    def __init__(self, char: str, vowel: bool) -> None:
        self.char = char
        self.vowel = vowel

    def __repr__(self) -> str:
        return self.char

letters = [
    Letter('a', True),
    Letter('b', False),
    Letter('c', False),
    Letter('d', False),
    Letter('e', True),
]

sorted_letters = sorted(letters, key=lambda letter: letter.vowel)
grouped = [list(result) for key, result in groupby(
    sorted_letters, key=lambda letter: letter.vowel)]

print(grouped)
# => [[b, c, d], [a, e]]