[python] Peewee를 사용한 데이터 분석 작업

Peewee는 Python 프로그래밍 언어를 위한 간단하고 가벼운 ORM(Object-Relational Mapping) 라이브러리입니다. 이 라이브러리를 사용하면 데이터베이스와 상호작용하는 작업을 더욱 쉽게 할 수 있습니다.

이번 포스트에서는 Peewee를 사용하여 데이터 분석 작업을 수행하는 방법에 대해 알아보겠습니다.

1. Peewee 설치하기

Peewee를 사용하기 위해서는 먼저 설치해야 합니다. pip를 사용하여 아래의 명령어를 실행합니다.

pip install peewee

2. 데이터베이스 연결하기

Peewee는 다양한 데이터베이스 시스템과 호환됩니다. 데이터베이스에 연결하기 위해 Database 클래스를 사용합니다. 아래의 예제는 SQLite 데이터베이스에 연결하는 방법입니다.

from peewee import *

db = SqliteDatabase('data.db')

3. 모델 정의하기

Peewee는 데이터베이스 테이블을 모델로 표현합니다. 각 테이블은 하나의 모델 클래스로 정의됩니다. 모델 클래스는 Model 클래스를 상속받아 만듭니다.

class User(Model):
    name = CharField()
    age = IntegerField()

    class Meta:
        database = db

4. 테이블 생성하기

테이블을 생성하기 위해 create_table() 메서드를 사용합니다. 아래의 예제는 User 모델에 해당하는 테이블을 생성하는 방법을 보여줍니다.

db.connect()
db.create_tables([User])

5. 데이터 삽입하기

테이블에 데이터를 삽입하기 위해 create() 메서드를 사용합니다. 아래의 예제는 User 테이블에 데이터를 삽입하는 방법을 보여줍니다.

user = User.create(name="John", age=25)

6. 데이터 조회하기

테이블의 데이터를 조회하기 위해 select() 메서드를 사용합니다. 아래의 예제는 User 테이블의 모든 데이터를 조회하는 방법을 보여줍니다.

users = User.select()
for user in users:
    print(user.name, user.age)

7. 데이터 업데이트하기

테이블의 데이터를 업데이트하기 위해 update() 메서드를 사용합니다. 아래의 예제는 User 테이블의 데이터를 업데이트하는 방법을 보여줍니다.

user = User.get(User.name == "John")
user.age = 30
user.save()

8. 데이터 삭제하기

테이블의 데이터를 삭제하기 위해 delete_instance() 메서드를 사용합니다. 아래의 예제는 User 테이블에서 이름이 “John”인 데이터를 삭제하는 방법을 보여줍니다.

user = User.get(User.name == "John")
user.delete_instance()

마무리

Peewee를 사용하여 데이터 분석 작업을 수행하는 방법에 대해 알아보았습니다. Peewee는 간편한 문법과 다양한 기능을 제공하여 데이터베이스와의 상호작용을 편리하게 할 수 있습니다. 자세한 내용은 Peewee 공식 문서를 참고하시기 바랍니다.

Happy coding!