Python - 객체 리스트 정렬

Integer, String 같은 기본 자료형 외에, 객체들을 갖고 있는 리스트 정렬 방법을 소개합니다.

1. list.sort()로 객체 리스트 정렬

아래 예제에서는 name과 age 값을 갖고 있는 Person 클래스를 정의하였습니다.

그리고 list.sort()로 리스트를 정렬합니다. sort()는 인자로 람다식을 받으며, 어떤 데이터를 기준으로 객체를 정렬할 지 대상을 설정할 수 있습니다.

  • sort(key=lambda x: x.name) : name으로 오름차순 정렬
  • sort(key=lambda x: x.name, reverse=True) : name으로 내림차순 정렬
  • sort(key=lambda x: x.age) : age로 오름차순 정렬
  • sort(key=lambda x: x.age, reverse=True) : age로 내림차순 정렬
class Person:
    def __init__(self, name, age):
        self.name = name
        self.age = age

    def __repr__(self):
        return '{' + self.name + ', ' + str(self.age) + '}'


if __name__ == '__main__':
    people = [
        Person('John', 35),
        Person('Patrick', 27),
        Person('Alex', 31)
    ]
    # sort by name (ascending)
    people.sort(key=lambda x: x.name)
    print(people)

    # sort by name (descending)
    people.sort(key=lambda x: x.name, reverse=True)
    print(people)

    # sort by age (ascending)
    people.sort(key=lambda x: x.age)
    print(people)

    # sort by age (descending)
    people.sort(key=lambda x: x.age, reverse=True)
    print(people)

Output:

[{Alex, 31}, {John, 35}, {Patrick, 27}]
[{Patrick, 27}, {John, 35}, {Alex, 31}]
[{Patrick, 27}, {Alex, 31}, {John, 35}]
[{John, 35}, {Alex, 31}, {Patrick, 27}]

2. sorted()로 객체 리스트 정렬

위의 방법은 원본 리스트의 순서를 변경합니다.

만약 원본 리스트는 변경하지 않고, 정렬된 새로운 리스트를 리턴하고 싶다면 sorted()를 사용할 수 있습니다.

sort()와 유사하게, sorted(list, key, reverse=False)로 인자를 받고 정렬된 리스트를 리턴합니다. 기본적으로 오름차순으로 정렬하며, 내림차순으로 정렬하고 싶을 때는 reverse=True를 인자로 전달하면 됩니다.

class Person:
    def __init__(self, name, age):
        self.name = name
        self.age = age

    def __repr__(self):
        return '{' + self.name + ', ' + str(self.age) + '}'


if __name__ == '__main__':
    people = [
        Person('John', 35),
        Person('Patrick', 27),
        Person('Alex', 31)
    ]
    # sort by name (ascending)
    result = sorted(people, key=lambda x: x.name)
    print(result)

    # sort by name (descending)
    result = sorted(people, key=lambda x: x.name, reverse=True)
    print(result)

    # sort by age (ascending)
    result = sorted(people, key=lambda x: x.age)
    print(result)

    # sort by age (descending)
    result = sorted(people, key=lambda x: x.age, reverse=True)
    print(result)

Output:

[{Alex, 31}, {John, 35}, {Patrick, 27}]
[{Patrick, 27}, {John, 35}, {Alex, 31}]
[{Patrick, 27}, {Alex, 31}, {John, 35}]
[{John, 35}, {Alex, 31}, {Patrick, 27}]
Loading script...

Related Posts

codechachaCopyright ©2019 codechacha