[python] 파이썬으로 MongoDB의 데이터 전처리하기

21 Nov 2023

python

이번 포스트에서는 파이썬을 사용하여 MongoDB의 데이터를 전처리하는 방법에 대해 알아보겠습니다.

1. MongoDB와 PyMongo

MongoDB는 NoSQL 데이터베이스로서, JSON 형태로 데이터를 저장하고 관리합니다. 파이썬에서 MongoDB와 상호 작용하기 위해서는 PyMongo 모듈을 사용합니다.

PyMongo를 설치하기 위해 다음 명령어를 실행합니다:

pip install pymongo

2. 데이터 가져오기

PyMongo를 사용하여 MongoDB에서 데이터를 가져오려면 다음 단계를 따릅니다:

from pymongo import MongoClient

# MongoDB와 연결
client = MongoClient('<mongodb 연결 문자열>')

# 데이터베이스 선택
db = client['<데이터베이스 이름>']

# 컬렉션 선택
collection = db['<컬렉션 이름>']

# 데이터 가져오기
data = collection.find()

위 코드에서 <mongodb 연결 문자열>, <데이터베이스 이름>, <컬렉션 이름>은 사용자의 환경에 맞게 수정해야 합니다.

3. 데이터 전처리

데이터를 가져온 후에는 필요한 전처리 작업을 수행할 수 있습니다. 예를 들어, 데이터의 필드를 수정하거나 정렬할 수 있습니다.

# 필드 수정
for item in data:
    item['field'] = item['field'].lower()

# 정렬
sorted_data = sorted(data, key=lambda x: x['field'])

위 코드에서는 데이터의 ‘field’ 필드를 모두 소문자로 변경하고, ‘field’ 필드를 기준으로 데이터를 정렬하는 예시입니다.

4. 데이터 업데이트

전처리가 완료된 데이터를 다시 MongoDB에 업데이트하려면 다음과 같은 코드를 사용할 수 있습니다:

# 데이터 업데이트
for item in sorted_data:
    collection.update_one({'_id': item['_id']}, {'$set': item})

위 코드에서 sorted_data는 전처리가 완료된 데이터를 의미하며, ‘_id’ 필드를 기준으로 MongoDB의 해당 문서를 찾아 업데이트합니다.

5. 결론

이번 포스트에서는 파이썬을 사용하여 MongoDB의 데이터를 전처리하는 방법에 대해 알아보았습니다. PyMongo를 사용하여 데이터를 가져오고, 필요한 전처리 작업을 수행한 후에는 다시 MongoDB에 업데이트할 수 있습니다. MongoDB와 PyMongo의 다양한 기능을 활용하여 데이터 전처리 작업을 더욱 효율적으로 수행해 보세요.

참고 자료

PyMongo 공식 문서: https://pymongo.readthedocs.io/
MongoDB 공식 문서: https://docs.mongodb.com/