[Python]파이썬으로 CSV와 JSON 파일 다루기: 입출력 가이드

데이터 처리 및 파일 입출력은 프로그래밍에서 중요한 부분입니다. 특히 CSV와 JSON은 데이터를 효율적으로 저장하고 교환하는 데 널리 사용됩니다. 이번 포스트에서는 python을 사용하여 CSV와 JSON 파일을 다루는 방법에 대해 알아보겠습니다.

≣ 목차

• csv 파일 입출력📂
• json파일 입출력📁
• 응용 예제🎁
• 핵심 내용👀

---

1. csv 파일 입출력📂

csv 파일은 텍스트 기반의 데이터를 행과 열로 구분하여 저장하는 형식입니다. csv 파일을 읽은 후 파일 안에 있는 내용을 출력하는 코드를 확인해 보겠습니다.

# csv 파일 읽기
student_list = []  # 최종 결과를 담을 빈 리스트를 만든다

# './csvfile/학생명단.csv' 경로의 파일을 읽기 모드('rt')로 열고 'file' 변수에 할당한다
with open('./csvfile/학생명단.csv', 'rt') as file:
    file.readline()  # 상단의 제목 줄을 읽어 건너뛴다
    while True:  # 무한 루프 시작
        line = file.readline()  # 한 줄씩 읽어 line 변수에 할당한다
        if not line:  # 만약 더 이상 읽을 라인이 없다면
            break  # 루프를 종료한다
        student = line.split(',')  # ,로 구분된 문자열을 분리하여 리스트 'student'에 저장한다
        student_list.append(student)  # 분리한 학생 정보를 결과 리스트에 추가한다

print(student_list)  # 결과 리스트를 출력한다

위 코드에서 읽은 CSV 파일의 내용

먼저 주어진 경로에 있는 csv 파일을 열고, csv 파일의 첫 번째 줄(제목 줄)을 건너뛴 후 나머지 내용을 읽어와서 각 줄을 쉼표로 구분하여 리스트에 저장하였습니다. 그 결과로 얻은 리스트를 출력합니다.

파이썬에서 파일을 읽어오는 방법에 대해 궁금하시다면 지난 파일 입출력 포스팅을 확인해 주세요😁

[Pytohon]파이썬 파일 입력 출력: 데이터 읽기와 쓰기의 기초

파이썬에서는 csv모듈을 사용하여 csv 파일을 쉽게 읽고 쓸 수도 있습니다. 다음은 동일한 csv 파일을 csv 모듈을 이용하여 읽고 쓰는 간단한 예제 코드입니다.

import csv

# csv 파일 읽기
student_list = []  # 최종 결과를 담을 빈 리스트를 만든다

# './csvfile/학생명단.csv' 경로의 파일을 읽기 모드('rt')로 열고 'file' 변수에 할당한다
with open('./csvfile/학생명단.csv', 'rt', newline='') as file:
    csv_reader = csv.reader(file)  # CSV 리더 객체 생성
    next(csv_reader)  # 첫 번째 줄(제목 줄)을 건너뛴다
    for line in csv_reader:  # CSV 파일의 각 줄에 대해 반복
        student_list.append(line)  # 각 줄을 결과 리스트에 추가한다

# csv 파일 쓰기 (데이터 추가)
new_data = ["세종대왕", "010-6666-6666"]

# 추가한 데이터를 결과 리스트에 추가
student_list.append(new_data)

# 결과 리스트를 출력한다
print(student_list)

---

2. json파일 입출력📁

JSON 파일은 데이터를 텍스트로 표현하는 경량의 데이터 교환 형식입니다. 파이썬에서는 json 모듈을 사용하여 JSON 파일을 쉽게 읽고 쓸 수 있습니다. 다음은 JSON 파일을 생성하는 예제 코드입니다.

import json

# 딕셔너리 리스트 생성
dict_list = [
    {
        'name': 'emma',  # 학생 이름
        'school': 'Central High School',  # 학교 이름
        'grade/class': [3, 1]  # 학년/반
    },
    {
        'name': 'alice',  # 학생 이름
        'school': 'Seoul Girls Middle School',  # 학교 이름
        'grade/class': [2, 2]  # 학년/반
    }
]

# 딕셔너리 리스트를 JSON 문자열로 변환(인코딩)
json_string = json.dumps(dict_list)

# JSON 파일로 저장
with open('./jsonfile/dictList.json', 'wt') as file:
    file.write(json_string)  # JSON으로 인코딩한 내용을 JSON 파일에 쓴다

print('dictList.json 파일이 생성되었습니다.')

주어진 딕셔너리 리스트를 JSON 형식의 문자열로 변환하고, 그 문자열을 JSON 파일에 씁니다. 먼저 json.dumps()를 사용하여 딕셔너리 리스트를 JSON 형식으로 변환하고, 이를 파일에 씁니다. 그 후에 출력문을 통해 파일이 생성되었음을 알려줍니다.

# json 파일 읽기
import json

with open('./jsonfile/dictList.json', 'rt') as file:
    json_reader = file.read()  # 파일 전체 읽어 json_reader에 저장
    dict_list = json.loads(json_reader)  # JSON을 파이썬 객체로 변환 (디코딩)

# 변환된 객체를 순회하며 각 딕셔너리의 정보를 출력
for dic in dict_list:
    print('이름: {}'.format(dic['name']))
    print('학교: {}'.format(dic['school']))
    print('학년/반: {}'.format(dic['grade/class'][0]), end=' / ')
    print(dic['grade/class'][1])
    print('----------------')

이번에는 주어진 경로에 있는 JSON 파일을 읽어서 파이썬 객체로 변환하고, 변환된 객체를 순회하며 각 항목을 출력합니다.

---

3. 응용 예제🎁

아래는 사용자로부터 파일명을 입력받아서 해당 파일을 복사하는 코드입니다. 입력된 파일이 'txt' 확장자가 아닐 경우 오류 메시지를 출력하고, 'txt' 확장자를 가진 파일일 경우 해당 파일을 읽어와서 복사본 파일에 씁니다. 마지막으로 복사본 파일이 생성되었음을 사용자에게 알려줍니다.

while True:
    filename = input('복사할 파일명을 입력하세요 >>> ')  # 사용자로부터 복사할 파일명을 입력 받는다
    extname = filename[filename.rfind('.') + 1:]  # 파일명에서 확장자를 추출한다

    # 확장자가 'txt'가 아니면 오류 메시지를 출력하고 다시 파일명을 입력받는다
    if extname != 'txt':
        print('복사할 수 없는 파일입니다.')
    else:
        break  # 'txt' 확장자인 경우 루프를 종료하고 파일 복사를 진행한다

# 원본 파일을 읽어와서 복사본 파일에 쓰는 작업을 수행한다
with open(filename, 'rt') as source:  # 원본 파일을 읽기 모드('rt')로 열어서 source 변수에 할당한다
    with open('복사본-' + filename, 'wt') as copy:  # 복사본 파일을 쓰기 모드('wt')로 열어서 copy 변수에 할당한다
        while True:
            buffer = source.read(1)  # 원본 파일에서 1바이트씩 읽어 buffer에 저장한다

이 코드는 서울특별시 강남구에 설치된 cctv의 총개수를 세는 기능을 수행합니다. CSV 파일을 읽고, 각 행의 7번째 열에 있는 값(설치된 cctv 개수)을 합산하여 출력합니다.

import csv

# './csvfile/cctv.csv' 파일을 읽기 모드('rt')로 열고 파일 객체를 csvfile에 할당한다
with open('./csvfile/cctv.csv', 'rt') as csvfile:
    # csv.reader를 사용하여 csvfile을 읽는다. delimiter는 쉼표(','), quotechar는 큰 따옴표('"')로 설정한다.
    buffer = csv.reader(csvfile, delimiter=',', quotechar='"')
    totalcctv = 0  # cctv 개수를 저장할 변수를 초기화한다

    # enumerate() 함수를 사용하여 버퍼의 각 행에 대해 반복한다
    for i, line in enumerate(buffer):
        if i != 0:  # 첫 번째 줄은 제목이므로 제외한다
            totalcctv += int(line[6])  # 각 행의 7번째 열(인덱스 6)에 있는 값(설치된 cctv 개수)을 totalcctv에 더한다

# 설치된 cctv의 총 개수를 출력한다
print('서울특별시 강남구에 설치된 cctv는 총 {}개입니다.'.format(totalcctv))

이번에는 'cctv.json' 파일에서 읽은 데이터를 파이썬 객체로 변환하고, 각 장소(place)의 '설치목적구분'을 세트에 추가합니다. 그 결과 중복되지 않는 목적 구분이 세트에 담겨 출력됩니다.

import json

# './jsonfile/cctv.json' 파일을 읽기 모드('rt')로 열고 파일 객체를 jsonfile에 할당한다
with open('./jsonfile/cctv.json', 'rt', encoding='utf-8') as jsonfile:
    buffer = jsonfile.read()  # 파일 내용을 문자열로 읽어온다
    cctv_list = json.loads(buffer)  # JSON 문자열을 파이썬 객체로 변환한다
    cctv_purpose = set()  # 결과를 담을 빈 세트를 생성한다 (중복 제외하기 위해)

    # cctv_list의 각 장소(place)에 대해 반복한다
    for place in cctv_list:
        cctv_purpose.add(place['설치목적구분'])  # 각 장소의 '설치목적구분'을 세트에 추가한다 (중복된 값은 추가되지 않는다)

print(cctv_purpose)  # 설치된 cctv의 목적 구분을 출력한다

---

4. 핵심 내용👀