파일 읽고 쓰기

Table of contents

Accessing a file
Reading a file
Writing a file
1. 내용 append하기
r+, w+, a+

Accessing a file

우선 open('file_name', 'mode')로 파일에 access
‘mode’의 종류: ‘r’: read, ‘w’: write, ‘a’: append

# 파일이 같은 폴더에 있을 경우
f = open('example.txt', 'r')  ## 이렇게 파일명만 적어주면 됨

# 파일이 다른 폴더에 있을 경우
f = open('C:/Users/admin/Desktop/chaelist_blog/example.txt', 'r')  ## 절대 경로로 접근

# 파일이 현재 python 파일을 실행시킨 폴더 내 다른 폴더에 있을 경우
f = open('data/example.txt', 'r')  ## data라는 폴더 내에 파일이 있는 경우

# 파일이 현재 python 파일을 실행시킨 폴더 밖의 폴더에 있을 경우
f = open('../example.txt', 'r')  ## ../를 통해 traverse up the directory

Reading a file

f.read()

: 파일 전체를 하나의 string value로 읽어준다

f = open('example.txt', 'r')   # 'r'은 default mode이기 때문에 안써줘도 결과는 동일
f.read()   # 아래처럼, 하나의 string value로 읽어옴

'1\n2\n3\n4\n5'

f.close()  # 특정 파일을 open한 다음 read / write하고 나면 닫아줘야 한다

f.readlines()

: read the entire file, as a list type (각 줄을 각각의 list variable로 받아서, 하나의 list로 반환)

가장 흔히 사용하는 file reading method.

f = open('example.txt', 'r')
f.readlines()   # 각 줄을 element로 하는 list를 반환

['1\n', '2\n', '3\n', '4\n', '5']

f.close()

f.readline()

: read the file line by line

한줄씩 차례대로 readline해줘야 해서, 잘 안 쓰는 method.

f = open('example.txt', 'r')
f.readline()   # 이런 식으로 한줄씩 읽힌다

'1\n'

f.readline()   # 이런 식으로 한줄씩 읽힌다

'2\n'

f.close()

활용 Tip

보통, readlines()에 for문을 결합해서 이용한다.

 f = open('example.txt', 'r')
 for line in f.readlines():
     print(line.strip())   # 보통 이렇게 for문을 사용해서 한 줄 한 줄 접근

 f.close()

readlines()를 쓰지 않아도, for문을 사용하면 한 줄 한 줄 접근이 가능하다

 f = open('example.txt', 'r')
 for line in f:
     print(line.strip())
        
 f.close()

txt 파일에서 읽어온 값들은 일단 다 string 형태로 저장된다.

 f = open('example.txt', 'r')
 lines_list = f.readlines()  # 이렇게 저장해두면 불러오기 편하다
 f.close()

 line = lines_list[0].strip()

 print(line, type(line))  # type을 출력해보면, txt 파일에서 읽어온 value들은 다 일단 string 형태.

 num_line = int(line)  # '' 사이의 값이 숫자일 경우, 이렇게 int()를 적용해  숫자로 바꿔줄 수 있다
 print(num_line, type(num_line))  # 이제 type이 integer로 바뀌어서, 사칙연산 가능

 1 <class 'str'>
 1 <class 'int'>

file open & close를 한 줄로 해결하는 방법:
```
 with open('파일명.txt', 'r', encoding='utf-8') as file:
     file.read()
```
- 이렇게 써주면 open()과 close()를 모두 써준 것과 동일
  (with 아래의 코드가 모두 실행되고 나면 알아서 파일이 닫힌다)
커서 옮기기
(※ 파일을 읽고 쓰는 과정에서 일종의 ‘커서’가 계속 줄을 따라 이동하게 되기 때문에, 커서 위치를 유의해야 할 때가 있음)
1. f.readline()으로 한 줄을 이미 읽어준 후라서, 커서의 위치는 두번째 줄로 이동함. 그 상태에서 f.read()로 읽어주면 커서 위치부터 가장 끝까지의 내용이 출력됨.
```
 f = open('sample.txt')
 print(f.readline())
 print('---------')
 print(f.read())
```
```
 1

 ---------
 2
 3
 4
 5
```
2. seek()을 활용하면 커서 위치를 옮길 수 있음 (seek(0): 파일의 가장 처음으로 커서를 옮겨줌)
```
 f = open('sample.txt')

 print(f.readline())
 print('---------')
 f.seek(0)
 print(f.read())
```
```
 1

 ---------
 1
 2
 3
 4
 5
```

Writing a file

f.write('내용')의 형태로 작성
다 쓴 file은 close해주어야 제대로 정보가 저장된다

# 예시
f1 = open('name.txt', 'w') ## 해당 python 파일이 실행된 동일한 폴더에 저장됨
f1.write('Chung & Cho')  # 쓰고 싶은 내용을 작성
f1.close()

write하는 내용은 무조건 string value여야 한다!

 f2 = open('numbers.txt', 'w')
 num = 3
 f2.write(num)  # 이렇게 int 형태의 데이터를 write하려고 하면 error
 f2.close

 ---------------------------------------------------------------------------
 TypeError                                 Traceback (most recent call last)
 <ipython-input-15-7b27a5c24a65> in <module>()
     1 f2 = open('numbers.txt', 'w')
     2 num = 3
 ----> 3 f2.write(num)
     4 f2.close

 TypeError: write() argument must be str, not int

cf) int 형태의 데이터는 str(num)으로 string으로 바꿔주면 write 가능

 f2 = open('numbers.txt', 'w')
 num = 3
 f2.write(str(num))
 f2.close()

한글 처리 방법: encoding='utf-8'

f3 = open('한글.txt', 'w', encoding='utf-8')  # encoding 방법을 지정 가능
f3.write('김태희')
f3.close()

여러 내용 이어서 write

 f_test = open('test.txt', 'w')
 f_test.write('test1'+'\n')
 f_test.write('test2')  # 이런식으로 이어서 f.write()를 또 해주면 계속 이어서 써짐
 ## f.close()를 해주기 전까지는 계속해서 한 줄씩 추가 가능

 f_test.close()  

내용 append하기

(이미 만든 파일에 내용 추가로 더 쓰기)

f_test = open('test.txt', 'a')  # 'a': append
f_test.write('\n'+'test3') # 'a'로 open한다음에 write 해주면 이미 존재하는 파일에 이어서 쓸 수 있음.
f_test.close() 

※ open('이미 있는 파일명', 'w') 이렇게 이미 있는 파일을 ‘w’(write)로 열면 그 위에 overwrite하게 되기 때문에, 이어서 계속 쓰고 싶다면 ‘a’(append)로 열어줘야 한다

r+, w+, a+

open() 함수에는 +모드들도 존재. 읽기 & 쓰기 기능을 한번에 처리하고 싶은 경우에 유용하다
총 6개의 모드: r, w, a, r+, w+, a+

r+ 모드

r 모드로 열면 reading만 가능하지만, r+ 모드로 열면 reading과 writing이 모두 가능하다
- ‘r’ 모드로 열고 ‘write’ 함수를 사용하면 ‘UnsupportedOperation: not writable’이라고 error가 남
r+ 모드로 열고 ‘write’하는 경우, 새로 적는 내용은 커서 위치부터 작성된다. 파일을 연 직후에는 커서 위치가 0 (가장 앞)이므로, 그 상태에서 write하는 경우 파일의 맨 처음부터 새로운 내용이 덮어씌워진다
- ※ r+ 모드로 열어서 먼저 ‘read’를 한 번 하고 나서 ‘write’를 하는 경우, 커서가 가장 뒤에 가 있으므로 뒤에 이어서 새로운 내용을 추가할 수 있다

# 샘플 파일 준비
with open('sample.txt', 'r') as f:
  print(f.read())

first line
second line

with open('sample.txt', 'r+') as f:
    ## 새로운 내용이 파일의 가장 앞부분부터 덮어씌워짐
    f.write('write new text')

    f.seek(0)  # 처음부터 읽어주기 위해서는 커서 위치를 0으로 바꿔줘야 함
    print(f.read())
  

write new textcond line

w+ 모드

w+ 모드로 열면 기존 내용을 지우고 새롭게 write하는 것 외에도, read 기능들도 사용 가능하다 (ex. 내용을 적고 바로 확인하는 것이 가능)

# 샘플 파일 준비
with open('sample.txt', 'r') as f:
  print(f.read())

first line
second line

with open('sample.txt', 'w+') as f:
    # 기존 내용은 지워지고, 새로운 내용이 적힌다
    f.write('write new text')
    
    f.seek(0)
    print(f.read())

write new text

+) ‘w+’ 모드로 여는 순간 file은 백지화된다. w+ 모드로 열자마자 f.read()으로 읽어들이는 것은 의미가 없음

with open('sample.txt', 'w+') as f:
print(f.read())
# 아무것도 출력되지 않음

a+ 모드

a+ 모드로 열면 기존 내용에 새롭게 내용을 덧붙여 write하는 것 외에도, read 기능들도 사용 가능하다

# 샘플 파일 준비
with open('sample.txt', 'r') as f:
    print(f.read())

first line
second line

with open('sample.txt', 'a+') as f:
    # 파일의 기존 내용은 유지하고, 맨 뒤에 새로운 내용이 추가된다
    f.write('\nwrite new text')

    # write한 후의 text 내용 출력
    f.seek(0)
    print(f.read())

first line
second line
write new text

+) ‘a+’ 모드로 열면 커서 위치가 파일 맨 끝부터 시작한다. 그러므로 a+모드로 열자마자 read해주려면 seek(0)으로 커서 위치를 조정해줘야 한다

with open('sample.txt', 'a+') as f:
    print(f.read())
# 아무것도 출력되지 않음
# seek(0) 하고 f.read()를 해줘야 전체 내용을 출력 가능