[파이썬] 데이터 불러오기와 저장하기

01 Sep 2023

python

데이터는 많은 분석 작업에서 중요한 역할을 합니다. 파이썬은 데이터를 불러오고 저장하는 다양한 방법을 제공합니다. 이 블로그 포스트에서는 파이썬에서 데이터를 불러오고 저장하는 여러 방법을 살펴보겠습니다.

CSV 파일 불러오기와 저장하기

CSV(comma-separated values)는 데이터를 저장하는 일반적인 형식입니다. 파이썬에서는 csv 모듈을 사용하여 CSV 파일을 불러오고 저장할 수 있습니다.

CSV 파일 불러오기

import csv

# CSV 파일 불러오기
with open('data.csv', 'r') as file:
    csv_reader = csv.reader(file)
    for row in csv_reader:
        # 각 행의 데이터에 접근하여 원하는 작업 수행
        print(row)

CSV 파일 저장하기

import csv

data = [
    ['이름', '나이', '성별'],
    ['John', '25', '남성'],
    ['Alice', '30', '여성'],
    ['Bob', '35', '남성']
]

# CSV 파일 저장하기
with open('data.csv', 'w', newline='') as file:
    csv_writer = csv.writer(file)
    csv_writer.writerows(data)

Excel 파일 불러오기와 저장하기

데이터 분석에서 Microsoft Excel 파일은 자주 사용됩니다. 파이썬에서는 pandas 라이브러리를 사용하여 Excel 파일을 불러오고 저장할 수 있습니다.

Excel 파일 불러오기

import pandas as pd

# Excel 파일 불러오기
data = pd.read_excel('data.xlsx')

# 데이터 프레임 출력
print(data)

Excel 파일 저장하기

import pandas as pd

data = {
    '이름': ['John', 'Alice', 'Bob'],
    '나이': [25, 30, 35],
    '성별': ['남성', '여성', '남성']
}

# 데이터 프레임 생성
df = pd.DataFrame(data)

# Excel 파일 저장하기
df.to_excel('data.xlsx', index=False)

JSON 파일 불러오기와 저장하기

JSON(JavaScript Object Notation)은 데이터를 저장하는 또 다른 형식입니다. 파이썬에서는 json 모듈을 사용하여 JSON 파일을 불러오고 저장할 수 있습니다.

JSON 파일 불러오기

import json

# JSON 파일 불러오기
with open('data.json', 'r') as file:
    json_data = json.load(file)
    print(json_data)

JSON 파일 저장하기

import json

data = {
    '이름': 'John',
    '나이': 25,
    '성별': '남성'
}

# JSON 파일 저장하기
with open('data.json', 'w') as file:
    json.dump(data, file)

위에서 언급한 방법들을 사용하여 파이썬에서 데이터를 불러오고 저장할 수 있습니다. 데이터 형식에 따라 적절한 라이브러리나 모듈을 선택하여 작업하면 됩니다. 데이터를 효율적으로 저장하고 불러올 수 있다면, 데이터 분석 작업을 원활하게 수행할 수 있습니다.

이상으로 데이터 불러오기와 저장하기에 대한 파이썬의 다양한 기능을 살펴보았습니다. 이를 통해 더 효율적인 데이터 분석 작업을 진행할 수 있을 것입니다.