Python Academy · 강의

데이터 클래스 정의하기

상용구 코드를 줄이기 위해 @dataclass를 사용해 보세요.

데이터 클래스 정의하기은(는) CoddyKit의 무료 Python Academy 강의입니다. 이것은 4개 중 1번째 강의입니다. 아래에서 전체 강의를 무료로 읽을 수 있으며, 내장 코드 에디터와 24/7 AI 튜터와 함께 브라우저에서 직접 실습할 수 있습니다. 이 강의는 Python Academy 학습 경로의 일부이며, 진행 상황이 웹과 CoddyKit 앱에 동기화됩니다. Python Academy 강의에는 총 4개의 강의가 포함되어 있습니다.

데이터 클래스를 사용하는 이유

대부분 데이터를 담는 클래스를 작성하면 같은 상용구를 반복하게 됩니다. __init__, __repr__, __eq__가 그 예입니다. dataclasses 모듈은 간단한 필드 선언만으로 이 모든 코드를 대신 생성합니다.

작성하고 읽어야 할 코드가 줄어듭니다.
__init__에서 오타가 발생할 가능성이 줄어듭니다.
Python 3.7부터 표준 라이브러리에 포함되어 있습니다.

기존 방식

다음은 점을 저장하는 일반 클래스입니다. 숫자 두 개를 저장하는 데만 얼마나 많은 코드를 입력해야 하는지 확인해 보십시오.

class Point:
    def __init__(self, x, y):
        self.x = x
        self.y = y

p = Point(1, 2)
print(p.x, p.y)

@dataclass 데코레이터

클래스 위에 @dataclass를 적용하고 형식 주석과 함께 필드를 선언합니다. Python이 __init__을 자동으로 생성합니다.

from dataclasses import dataclass

@dataclass
class Point:
    x: int
    y: int

p = Point(1, 2)
print(p.x, p.y)

repr 자동 생성

데이터 클래스는 읽기 쉬운 __repr__도 생성하므로, 객체를 출력하면 메모리 주소 대신 필드가 표시됩니다.

from dataclasses import dataclass

@dataclass
class Point:
    x: int
    y: int

print(Point(3, 4))

eq 자동 생성

모든 필드가 같으면 두 데이터 클래스 인스턴스가 동일한 것으로 비교됩니다. __eq__를 직접 작성할 필요가 없습니다.

from dataclasses import dataclass

@dataclass
class Point:
    x: int
    y: int

print(Point(1, 2) == Point(1, 2))
print(Point(1, 2) == Point(9, 9))

형식 주석은 필수입니다

각 필드에는 반드시 형식 주석이 있어야 합니다. 데이터 클래스의 내부 동작은 이 주석을 통해 해당 이름이 필드임을 알 수 있습니다. 주석 없이 단독으로 대입하면 필드가 아닌 일반 클래스 속성이 됩니다.

from dataclasses import dataclass

@dataclass
class User:
    name: str
    age: int

u = User('Alice', 30)
print(u)

메서드 추가

데이터 클래스도 여전히 일반 클래스입니다. 생성된 필드를 사용하는 일반 메서드를 추가할 수 있습니다.

from dataclasses import dataclass

@dataclass
class Rectangle:
    width: int
    height: int

    def area(self):
        return self.width * self.height

r = Rectangle(3, 4)
print(r.area())

필드 변경

기본적으로 데이터 클래스 인스턴스는 변경 가능하므로, 일반 속성과 마찬가지로 생성 후에도 필드를 다시 할당할 수 있습니다.

from dataclasses import dataclass

@dataclass
class Counter:
    value: int

c = Counter(0)
c.value += 5
print(c.value)

asdict 및 astuple

asdict()와 astuple() 도우미는 인스턴스를 딕셔너리나 튜플로 변환하므로 직렬화할 때 편리합니다.

from dataclasses import dataclass, asdict, astuple

@dataclass
class Point:
    x: int
    y: int

p = Point(1, 2)
print(asdict(p))
print(astuple(p))

중첩 데이터 클래스

데이터 클래스의 필드 자체가 또 다른 데이터 클래스일 수 있습니다. 생성된 __repr__과 asdict()는 중첩된 구조까지 재귀적으로 처리합니다.

from dataclasses import dataclass, asdict

@dataclass
class Address:
    city: str

@dataclass
class Person:
    name: str
    address: Address

p = Person('Bob', Address('Paris'))
print(asdict(p))

데이터 클래스를 사용할 때

클래스가 주로 서로 관련된 값을 함께 묶는 역할을 한다면 데이터 클래스를 사용하십시오.

구성 객체
함수에서 반환되는 레코드
좌표나 화폐처럼 단순한 값 형식

데이터보다 동작이 중심인 클래스라면 일반 클래스가 더 명확하게 읽힐 수 있습니다.

빠른 확인

@dataclass 데코레이터는 무엇을 자동으로 생성합니까?

복습

상용구를 없애기 위해 @dataclass를 사용하는 방법을 배웠습니다.

형식 주석과 함께 필드를 선언합니다.
__init__, __repr__, __eq__가 생성됩니다.
asdict()와 astuple()을 사용하여 인스턴스를 변환합니다.
메서드를 추가하고 데이터 클래스를 중첩할 수도 있습니다.

자주 묻는 질문