Contents

[Python] 파일 입출력

1. 파일 읽기

1
fileobj = open(filename, mode)
  • fileobj는 open() 함수에 의해 반환되는 객체
  • filename은 파일의 문자열 이름
  • mode는 파일의 종류와 읽기 모드를 의미
  • mode의 첫 번째 글자 의미
    • r: 파일 읽기
    • w: 파일 쓰기(파일이 없을 경우 생성, 있을 경우 덮어쓰기)
    • x: 파일 쓰기(파일이 있을 경우 덮어쓰지 않음, FileExist Error 발생)
    • a: 파일의 끝에서부터 이어서 쓰기
  • mode의 두 번째 글자 의미
    • t: text type
    • b: binary type
1
fileobj.close()
  • 파일을 열고 모든 작업 완료 후, close() 함수를 이용하여 파일을 닫아야 함

2.1 Text 파일 쓰기: write('filename', 'wt')

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
poem = '''There was a young lady named Bright,
Whose speed was far faster than light;
She started one day
In a relative way,
And returned on the previous night.'''

if __name__ == "__main__":
    # file open
    fout = open('relativity', 'wt')
    # write
    print(fout.write(poem))
    # file close
    fout.close()
  • write() 함수는 파일에 쓴 byte 수를 반환
  • 텍스트 파일의 경우 wt mode 사용
  • print() 함수를 이용한 파일 쓰기도 가능
    • print(poem, file=fout, sep='', end='')
  • 파일에 쓸 문자열의 크기가 크다면 특정 크기(chunk)만큼 쪼개어 파일 쓰기가 가능
  • 다음은 chunk 크기 단위로 파일 쓰기를 수행하는 예제코드이다.
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
poem = '''There was a young lady named Bright,
Whose speed was far faster than light;
She started one day
In a relative way,
And returned on the previous night.'''

if __name__ == "__main__":
    # 파일에 쓸 문자열이 클 경우 chunk 단위로 나누어서 파일에 쓸 수 있음
    fout = open('relativity3', 'wt')

    size = len(poem)
    offset = 0
    chunk = 100 # 쓰기 단위를 지정(문자 100개)

    while True:
        if offset > size:
            break
        fout.write(poem[offset:offset+chunk])
        offset += chunk

    # file close
    fout.close()

2.2 Binary 파일 쓰기: write('filename', 'wb')

1
2
3
4
5
6
7
# 0~255 256 bytes 값 생성
bdata = bytes(range(0, 256))
print(len(bdata))

fout = open('binaryfile', 'wb')
print(fout.write(bdata)) # write() 함수는 파일에 쓴 byte 수를 반환하기 때문에 bdata의 값을 정상적으로 쓴 경우 256이 출력된다.
fout.close()
  • 바이너리 파일의 경우 wb mode 사용
  • 위의 예제코드는 256바이트 크기의 바이너리 값을 생성한 뒤, 이를 bdata 파일에 쓰기를 수행한다.
  • 바이너리 파일에는 텍스트를 제외한 이미지, 음성, 영상 등의 파일이 포함된다.

3.1 Text 파일 읽기: read(), readline(), readlines()

1
2
3
4
5
# read() 함수 예제코드
readfile = open('relativity', 'rt')
poem = readfile.read()
readfile.close()

  • read() 함수를 이용하여 파일을 읽을 수 있다. 인자 값이 없는 경우 파일 전체를 읽어 들인다.
  • read() 함수는 파일의 끝에 도달하면 빈 문자열('')을 반환하며 이는 False 값을 의미한다.
  • 텍스트 파일의 읽기는 rt mode 사용
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
# read(argument) 인자 값을 전달하는 read() 함수 예제코드
poem = ''
readfile = open('relativity', 'rt')
chunk = 100

while True:
    fragment = readfile.read(chunk) # read() 함수는 파일의 끝까지 다 읽을 경우 '' 빈 문자열을 반환
    if not fragment: # 더이상 읽은 내용이 없으면(read 함수의 반환 값이 빈 문자열) break --> 빈 문자열은 False
        break
    poem += fragment

readfile.close()

  • 인자 값을 통해 read() 함수가 읽어 들일 수 있는 크기를 제한할 수 있다.
  • 위의 예제에서는 chunk라는 변수를 통해 한 번에 읽어 들일 문자의 수를 제어한다.
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
# readline() 함수 예제코드
poem = ''
readfile = open('relativity', 'rt')
while True:
    line = readfile.readline() # readline() 함수는 파일을 한 라인씩 읽어 들인다.
    if not line: # 파일의 끝에 도달했을 경우 while문 종료
        break
    poem += line

readfile.close()
  • readline() 함수는 파일을 라인 단위로 읽는다.
  • read() 함수와 동일하게 파일의 끝에 도달하면 빈 문자열을 반환한다.
 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
# readlines() 함수 예제코드
readfile = open('relativity', 'rt')
lines = readfile.readlines() # 파일의 각 라인을 읽어들여 리스트의 각 요소로 저장
readfile.close()

print(len(lines)) # 라인 수 출력

for line in lines: # 각 라인이 저장된 리스트 lines의 항목을 하나씩 읽어서 출력하는 for문
    print(line, end="")

# 실행결과
# 5
# There was a young lady named Bright,
# Whose speed was far faster than light;
# She started one day
# In a relative way,
# And returned on the previous night.

  • readlines() 함수는 모든 라인을 읽은 뒤, 이를 리스트로 반환한다.
  • 파일의 각 라인이 리스트의 각 요소 값으로 저장된다.

3.2 Binary 파일 읽기: read(), readline(), readlines()

1
2
3
4
5
6
# Binary 파일의 read() 함수 호출 예제코드
readfile = open('bfile', 'rb') # rb 모드로 열기
bdata = readfile.read()
print(len(bdata))
readfile.close()

  • 텍스트 파일에서의 read() 함수와 동작은 동일하다.
  • 바이너리 파일의 읽기는 rb mode를 사용한다는 점에서 텍스트 파일 읽기와 차이가 있다.
  • readline(), realines() 사용 방법도 텍스트 파일 읽기와 동일하다.


본 포스팅 내용은 처음시작하는 파이썬(Introducing Python)을 학습한 내용에 대한 요약 글입니다. 코드에 문제가 있을 수 있으며 다소 부정확한 내용이 있을 수 있으니 참고하시기 바랍니다. 잘못된 내용이 있는 경우 덧글로 남겨주시면 내용을 반영하여 수정하도록 하겠습니다. 감사합니다.



[Reference]

[1] 처음시작하는 파이썬(Introducing Python), p.231-240