파이썬 데이터 클래스: 모던 문법으로 데이터 관리 간소화

파이썬 데이터 클래스는 @dataclass 데코레이터 가진 일반 파이썬 클래스. 데이터 보유 목적으로 특별히 생성. 파이썬 3.7부터 dataclass 내장 모듈로 데이터 클래스 제공. 일반 파이썬 클래스 대비 여러 이점, 이 글에서 탐구. 예제 코드와 데이터 클래스로 수행하고픈 흔한 연산 두어 가지도 살펴봄.

목차

데이터 클래스 사용 이점

왜 일반 파이썬 클래스 대신 데이터 클래스 써야 할까? 먼저 파이썬 데이터 클래스 제공 이점 몇 가지 보자.

최소한의 코드 필요

@dataclass 데코레이터가 눈에 보이는 코드 추가 없이 클래스에 많은 기능 추가. 데이터 클래스 매우 콤팩트하면서 많은 유용 기능 제공. 데이터 보유할 필드만 정의하면 됨. 어떤 함수도 정의할 필요 없음.

비교

두 파이썬 데이터 클래스는 ==로 비교 가능. 소위 던더 메서드 __eq__ 자동 구현되기 때문. 일반적으로 이 특수 메서드 구현한 파이썬 객체면 동일 타입 다른 객체와 비교 가능.

데이터 클래스 출력

마찬가지로 __repr__ 구현되어 데이터 클래스 출력하면 멋진 표현 나옴. 디버깅 시 특히 유용.

데이터 클래스는 타입 힌트 필요

데이터 클래스는 파이썬 제공 새로운(ish) 타입 시스템 중심으로 구축. 타입 힌트 사용으로 코드 버그·예상치 못한 동작 가능성 감소. 변수에 저장되어야 할 데이터 타입 본질적으로 선언.

데이터 클래스 예제

데이터 클래스 작동 예:

from dataclasses import dataclass
@dataclass
class Card:
    rank: str
    suit: str
card1 = Card("Q", "hearts")
card2 = Card("Q", "hearts")
print(card1 == card2)  # True
print(card1.rank)      # 'Q'
print(card1)           # Card(rank='Q', suit='hearts')

기본값

데이터 클래스 기본값 가질 수 있음. 기본값 할당은 변수에 값 할당처럼 간단. 예: Card 클래스 기본값을 하트 Q로 하려면:

from dataclasses import dataclass
@dataclass
class Card:
    rank: str = 'Q'
    suit: str = 'hearts'

데이터 클래스를 JSON으로 변환

흔한 유스케이스: 멋지게 구조화된 데이터 클래스 JSON 변환. 예: 데이터베이스 내보내기, 브라우저 전송. 나쁜 소식: 데이터 클래스 JSON 변환 내장 방식 없음.

좋은 소식: dataclasses-json 패키지 작업 단순화. 하지만 추가 데코레이터 필요. pip install이나 Pipenv로 패키지 설치, 가상환경 권장. 예:

$ pip install dataclasses-json

패키지 사용 예:

from dataclasses import dataclass
from dataclasses_json import dataclass_json
@dataclass_json
@dataclass
class Card:
    rank: str = 'Q'
    suit: str = 'hearts'
card = Card()
print(card.to_json())

{"rank": "Q", "suit": "hearts"}

다른 방법: 파이썬 상속으로 JSONEncoder 클래스 상속받아 커스텀 인코더 생성. 여기선 외부 패키지 설치 불필요. 이 블로그 포스트에서 방법 배울 수 있음.

계속 학습

• 파이썬 attrs 패키지는 네이티브 데이터 클래스 고급 버전
• Python.org 공식 문서
• 파이썬에서 여러 값 반환하는 법