[python] 파이썬에서 데이터 저장하는 방법

11 Dec 2023

python

파이썬을 사용하면 다양한 방식으로 데이터를 저장하고 관리할 수 있습니다. 이 글에서는 가장 일반적으로 사용되는 데이터 저장 방법에 대해 알아보겠습니다.

1. 텍스트 파일로 데이터 저장하기

가장 간단한 방법은 텍스트 파일에 데이터를 저장하는 것입니다. 이를 위해 파이썬에서는 파일을 열어 텍스트를 쓰거나 읽을 수 있는 기능을 제공합니다.

# 파일 쓰기
with open('data.txt', 'w') as file:
    file.write('저장할 데이터\n')

# 파일 읽기
with open('data.txt', 'r') as file:
    data = file.read()
    print(data)

2. CSV 파일로 데이터 저장하기

CSV(Comma-Separated Values) 파일은 표 형식의 데이터를 저장하는데 많이 사용됩니다. 파이썬에서는 내장 모듈 csv를 사용하여 CSV 파일을 쉽게 다룰 수 있습니다.

import csv

# CSV 파일 쓰기
data = [['이름', '나이'], ['철수', 30], ['영희', 28]]
with open('data.csv', 'w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(data)

# CSV 파일 읽기
with open('data.csv', 'r') as file:
    reader = csv.reader(file)
    for row in reader:
        print(row)

3. 데이터베이스에 저장하기

데이터베이스는 구조화된 데이터를 효율적으로 저장하고 관리하기 위한 도구입니다. 파이썬에서는 다양한 데이터베이스 시스템에 접속하여 데이터를 저장하고 조회할 수 있는 모듈을 제공합니다.

import sqlite3

# SQLite 데이터베이스에 연결
conn = sqlite3.connect('data.db')

# 테이블 생성
conn.execute('''
CREATE TABLE IF NOT EXISTS users (
    id INTEGER PRIMARY KEY,
    name TEXT NOT NULL,
    age INTEGER
);
''')

# 데이터 삽입
conn.execute('INSERT INTO users (name, age) VALUES (?, ?)', ('철수', 30))

# 데이터 조회
cursor = conn.execute('SELECT * FROM users')
for row in cursor:
    print(row)

# 연결 종료
conn.close()

위의 예시들을 통해 파이썬에서 데이터를 텍스트 파일, CSV 파일, 데이터베이스에 저장하는 방법에 대해 알아보았습니다. 데이터의 종류와 용도에 따라 적합한 저장 방법을 선택하여 활용하는 것이 중요합니다.

참고 자료: