파이썬으로 txt파일 만들고 읽기, write & read txt file

이번에는 파이썬에서 간단한 파일 IO처리를 알아보겠습니다.

프로그램에 사용하는 데이터는 프로그램이 실행되는 동안은 메모리에 저장이 되지만 프로그램이 종료되는 날라가 버립니다. 그래서 데이터를 보존할 방법이 필요한데 대량의 데이터를 처리를 위해서는 데이터베이스(database)가 필요하겠지만, 개인적인 작업을 저장해서 처리하거나 간단한 입출력을 위해서 database를 사용하는 것은 자원낭비일 수 있습니다. 이때 유용하게 활용할 수 있는 것이 파일로 저장하고 읽어서 사용하는 방법입니다.

fw = open('sample.txt', 'w')
fw.write('Writing some stuff in my text file\n')
fw.write('I like bacon\n')
fw.close()

첫번째 코드는 sample.txt라는 파일을 생성합니다. 그리고, 2줄의 내용을 파일에 씁니다. '\n' 이 부분 때문에 내용이 한줄로 이어지지 않고 줄바꿈이 된 것입니다.

코드를 보고 이미 짐작하셨겠지만, open() 함수가 바로 파일을 만들어 주는 파이썬 내장함수입니다.

첫번째 인수가 파일명이고, 두번째 인수는 파일을 open하는 방식입니다. 'w'는 해당 파일명이 존재하면 내용을 초기화(다 지우고)해서 open을 하고, 해당 파일명이 존재하지 않으면 새로 파일을 만들어서 open해줍니다.

기존 내용을 그대로 유지한 채로 open을 해야 할 경우는 'a'(append)로 지정하면 됩니다.

write() 함수는 파일에 내용을 쓰는 기능을 합니다. 파일에 쓰기가 완료되면 close() 함수를 호출해서 파일을 닫아주어야 합니다.

fr = open('sample.txt', 'r')
text = fr.read()
print(text)
fr.close()

이번에는 앞에서 만든 파일을 읽어드리겠습니다. 이미 존재하는 파일명을 첫번째 인수로 지정해 줘야 하고, 두번째 인수로 'r'을 지정하면 됩니다. 파일을 open한 후에는 read() 함수를 사용해서 내용을 읽을 수 있습니다.

읽기가 끝났으면 쓰기때와 마찬가지로 close() 함수를 호출해서 파일을 닫아 줘야 합니다.

파일을 닫아도 text변수에 파일의 내용을 읽어서 저장을 해뒀기 때문에 text변수에는 내용이 그대로 남아 있습니다.

이상의 간단한 파일 입출력 샘플을 참조해서 여러분이 수집한 자료를 저장하고 다시 읽어서 처리하는 작업에 활용하시면 매우 유용할 것입니다. 아직은 굳이 DB까지는 몰라도 되니까요~

--------------------

Source Code from thenewboston Tutorials

좋은 코드는 볼수록 Insight를 주고, 반복할수록 내 것과 같이 된다.