파이썬에 파일 입출력에 대해 공부한 부분을 정리해보려 한다.

파일 입출력

파이썬에서 텍스트 파일을 새로 만들거나 기존의 텍스트 파일을 불러들일 수 있다.
open() 메서드를 사용하여 파일을 열고, 작업이 끝나면 close() 메서드를 사용하여 파일을 닫아주어야 한다.

기본형태

변수 = open(파일명, 권한)
변수.close()

권한은 w a r 이 있다.
w : 파일을 새로 만들 때 사용
a : 기존 파일을 불러와 수정할 때 사용
r : 기존 파일을 읽을 때 사용
ps) 한글을 입력할 땐 문자 인코딩 작업을 해야한다. encoding 키워드를 사용해 utf8을 입력한다.

score_file = open("score.txt", "w", encoding="utf8")
score_file.close()

해당 코드를 입력하면 score.txt 파일이 생성된다.

쓰기

권한을 w 혹은 a 을 주고,
출력 키워드 file 를 사용하여 입력하거나
write 메서드를 사용하여 파일을 입력한다.

score_file = open("score.txt", "w", encoding="utf8")
print("수학 : 0", file=score_file)
print("영어 : 50", file=score_file)
score_file.close()

score_file = open("score.txt", "a", encoding="utf8")
score_file.write("과학 : 80")
score_file.write("\n코딩 : 100")
score_file.close()

해당 코드를 실행하면 core.txt 파일이 생성되고, 아래 내용이 채워져있다.

수학 : 0
영어 : 50
과학 : 80
코딩 : 100

읽기

권한을 r로 주고, read() 메서드를 사용하여 내용을 읽을 수 있다.

score_file = open("score.txt", "r", encoding="utf8")
print(score_file.read())
# 수학 : 0
# 영어 : 50
# 과학 : 80
# 코딩 : 100
score_file.close()

readline()

read() 메서드로 파일을 읽을 수 있지만, readline() 메서드를 사용하여 한 줄 씩 읽어 들일 수 있다.

score_file = open("score.txt", "r", encoding="utf8")
print(score_file.readline(), end="") # 수학 : 0
print(score_file.readline(), end="") # 영어 : 50
print(score_file.readline(), end="") # 과학 : 80
print(score_file.readline(), end="") # 코딩 : 100
score_file.close()

score_file = open("score.txt", "r", encoding="utf8")
while True:
  line = score_file.readline()
  if not line:
    break
  print(line, end="")
# 수학 : 0
# 영어 : 50
# 과학 : 80
# 코딩 : 100
score_file.close()

readlines()

readlines() 메서드를 사용하면 파일 내용을 리스트로 받아 올 수 있다.

score_file = open("score.txt", "r", encoding="utf8")
lines = score_file.readlines() # <class 'list'>
print(type(lines))
for line in lines:
  print(line, end="")
# 수학 : 0
# 영어 : 50
# 과학 : 80
# 코딩 : 100
score_file.close()

pickle

텍스트 파일이 아닌 데이터 자체를 파일 형태로 저장하는 방법이다. 사용을 위해서는 pickle 모듈을 임포트 해야한다.

import pickle

바이너리 파일로 변환하기 때문에 권한 설정이 이전과 다른 wb rb 로 설정한다.

쓰기

권한을 wb 로 주고, dump() 메서드를 사용하여 파일에 저장한다.

profile_file = open("profile.pickle", "wb")
profile = {"이름":"박명수", "나이":30, "취미":["축구", "골프"]}
pickle.dump(profile, profile_file)
profile_file.close()

해당 코드를 실행하면 pickle 확장자를 가진 파일이 생성된다.

읽기

권한을 rb 로 주고, load() 메서드를 사용하여 파일을 불러온다.

profile_file = open("profile.pickle", "rb")
profile = pickle.load(profile_file)
print(profile) # {'이름': '박명수', '나이': 30, '취미': ['축구', '골프']}
profile_file.close()

with

파일을 매번 쓰거나 불러들일 때 파일을 닫아줘야한다. 하지만 with 키워드를 사용하면 닫아주지 않아도 된다.
기본 형태

with open(파일명, 권한) as 변수:
  내용

권한 설정은 이전과 동일하다.

import pickle

with open("profile.pickle", "rb") as profile_file:
  print(pickle.load(profile_file))

with open("study.txt", "w", encoding="utf8") as study_file:
  study_file.write("파이썬을 열심히 공부하고 있어요")

with open("study.txt", "r", encoding="utf8") as study_file:
  print(study_file.read())

REFERENCE

나도코딩 유튜브
Python - 공식문서
점프투파이썬 - WikiDocs