인프런 커뮤니티 질문&답변

안녕하세요. 답변 도우미입니다.

1. r(읽기) 모드와 w(쓰기) 모드의 차이점

• r (read): 파일에서 데이터를 읽어오기 위해 열며, 파일이 이미 존재해야 합니다.

• w (write): 파일에 데이터를 쓰기 위해 열며, 파일이 존재하지 않으면 새로 만들고, 존재하면 내용을 전부 지운 뒤(truncate) 새로 쓰기 시작합니다.

즉, r 모드로 열면 기존 파일 내용을 읽을 수 있지만, w 모드로 열면 기존에 있던 내용이 모두 지워진 상태(또는 새 파일)에서 파일 핸들을 받게 됩니다.

2. DictReader가 동작하는 방식

import csv
​
with open('파일경로', 'r', encoding='utf-8-sig', newline='') as reader_csv:
    reader = csv.DictReader(reader_csv)
    for row in reader:
        print(row['first Name'], row['Last Name'])

위 코드에서 csv.DictReader는 CSV 파일의 첫 줄(헤더)을 읽어서 {'컬럼명': '값'} 형태의 딕셔너리를 만들어 줍니다.

• r 모드: 파일에 이미 내용(헤더와 데이터)이 있으므로, DictReader가 CSV 헤더를 찾아 딕셔너리를 잘 만들어냅니다.

• w 모드: 파일 내용을 전부 지워버린 뒤에 (혹은 새로 만든 상태에서) DictReader를 호출하게 되면, 헤더 자체가 없는 빈 파일이므로 DictReader가 인식할 컬럼명이 전혀 없습니다.

결과적으로 파일이 비어 있어서 DictReader에 의해 'first Name'이나 'Last Name'과 같은 키를 찾을 수 없게 되어 에러(KeyError 등)가 발생하거나, 내부적으로 파일 포인터가 읽을 내용을 찾지 못해 에러가 생기게 됩니다.

3. 해결 방법

1. 파일에서 읽어야 할 경우:

   • r 모드로 열고 DictReader 사용

   • 이미 존재하는 파일을 읽어서 무언가 처리가 필요하다면, 읽기 모드를 사용해야 함

2. 파일에 써야 할 경우:

   • w 모드로 열 때는, DictWriter 등을 사용해서 새로운 CSV 파일을 생성하겠다는 의도로 접근

   • 만약 기존 파일에 덧붙여 써야 한다면 a(append) 모드 사용 고려

예를 들어, “기존 파일을 읽어온 뒤 가공한 데이터를 다시 다른 파일로 출력”하려면 다음과 같이 코드를 분리해야 합니다.

import csv
​
# 읽기
with open('input.csv', 'r', encoding='utf-8-sig', newline='') as f:
    reader = csv.DictReader(f)
    rows = [row for row in reader]
​
# rows 처리 로직 ...
​
# 쓰기
with open('output.csv', 'w', encoding='utf-8-sig', newline='') as f:
    fieldnames = ['first Name', 'Last Name']
    writer = csv.DictWriter(f, fieldnames=fieldnames)
    writer.writeheader()
​
    for row in rows:
        writer.writerow({'first Name': row['first Name'],
                         'Last Name':  row['Last Name']})

위와 같이 “읽을 때는 읽기 모드(r)”, “쓸 때는 쓰기 모드(w)”로 다른 파일이든 같은 파일이든 구분하여 적절히 사용하면, DictReader/DictWriter를 올바로 활용할 수 있습니다.