파이썬으로 CSV 파일 다루기: 데이터 분석의 첫걸음

안녕하세요, 프로그래밍과 블로그를 동시에 즐기는 여러분! 오늘은 데이터 분석의 기초인 파일 처리와 CSV 파일을 다루는 내용을 살펴보겠습니다. 데이터 처리는 우리가 직면할 수 있는 다양한 문제를 해결하는 데 중요한 역할을 하며, 특히 데이터 세트를 분석하고 활용하는 방법은 여러 분야에서 필수적입니다. 이 포스트에서는 파이썬을 활용하여 CSV 파일을 읽고 쓰는 방법에 대해 알아보겠습니다.

들어가며

CSV(Comma-Separated Values) 파일은 데이터를 저장하는 데 널리 사용되는 형식입니다. 예를 들어, 학생 성적, 판매 기록, 고객 데이터 등 다양한 형태로 사용할 수 있죠. 파이썬은 이러한 CSV 파일을 효율적으로 처리하기 위한 다양한 라이브러리를 제공하므로, 매우 유용합니다.

자, 그럼 시작해볼까요?

코드 작성하기

파이썬의 내장 csv 모듈을 이용하여 CSV 파일을 읽고 쓰는 간단한 예제를 만들어 보겠습니다. 아래의 코드는 CSV 파일을 읽어와 내용을 출력하고, 새로운 데이터를 추가하여 새로운 CSV 파일로 저장하는 기능을 합니다.

import csv

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

# 새로운 데이터 추가 및 CSV 파일 쓰기
new_data = [['이름', '성적'], ['홍길동', 85], ['김유신', 90]]

with open('new_students.csv', mode='w', newline='') as file:
    writer = csv.writer(file)
    writer.writerows(new_data)

print("새로운 CSV 파일이 생성되었습니다.")

코드 설명하기

1. CSV 파일 읽기:
   • open() 함수를 사용하여 students.csv 파일을 읽는 모드로 열고, csv.reader()로 데이터를 읽습니다.
   • 각 행(row)의 데이터를 반복하여 출력합니다.
2. 새로운 데이터 추가 및 CSV 파일 쓰기:
   • 새로운 데이터는 리스트 형태로 정의하고, open() 함수로 new_students.csv 파일을 쓰기 모드로 엽니다.
   • csv.writer()를 사용하여 새로운 데이터를 CSV 형식으로 파일에 기록합니다.

예제 실행 결과

이 코드를 실행하면 students.csv의 내용이 출력되고, 새로운 학생 데이터가 new_students.csv로 저장됩니다. 예를 들어, students.csv가 다음과 같다면

이름,성적
홍길동,75
김유신,90

출력 결과는 다음과 같습니다:

['이름', '성적']
['홍길동', '75']
['김유신', '90']

그리고 new_students.csv 파일은 다음과 같은 내용을 갖게 됩니다:

이름,성적
홍길동,85
김유신,90

추가적으로

이 코드는 CSV 파일을 다루는 기본적인 방법을 소개한 것입니다. 여기에 몇 가지 확장 아이디어를 덧붙이면 좋겠습니다:

• 사용자에게 입력을 받아 파일명이나 데이터를 동적으로 관리하기.
• 데이터의 유효성을 검증하는 로직 추가.
• pandas와 같은 라이브러리를 사용하여 더 복잡한 데이터 처리 및 분석 가능하기.

마무리하며

이번 포스트에서는 파이썬을 통해 CSV 파일을 읽고 쓰는 방법을 알아보았습니다. 데이터 처리는 현대 사회에서 매우 중요하며, 이를 잘 활용하면 많은 문제를 해결하는 데 도움이 됩니다. 여러분도 직접 다양한 CSV 파일을 시도하고, 자신만의 데이터 분석 프로젝트를 만들어 보세요!

궁금한 점이나 추가 질문이 있으시면 언제든지 댓글로 남겨주세요. 다음 포스트에서 만나요!