- 0
- 이니스프리
- 조회 수 23364
파이썬으로 스크립트를 작성하다보면 텍스트 파일을 읽어서 각 행을 리스트 변수에 넣어야 하는 경우가 종종 있죠.
이하 filename.txt라는 파일의 각 행을 (line이란 변수에 넣어서) 최종적으로 list_file이란 리스트 변수에 대입하는 것으로
파일명과 변수명을 통일하여 설명해 보겠습니다 ^^
물론 다른 라이브러리를 사용하는 방법도 가능하지만 경우의 수가 너무 많아지니 배제할게요.
1. 다음과 같이 파일 객체를 readlines로 한 줄씩 읽고 각 줄의 끝에 붙은 '\n'을 지워주는 방법을 생각해볼 수 있습니다.
with open('filename.txt', 'r') as f: list_file = f.readlines() list_file = [line.rstrip('\n') for line in list_file]
다만 이 방법은 빅데이터처럼 아주 큰 파일을 다룰 때 readlines()가 효율적이지 않고
때로는 메모리 관련 에러를 유발할 소지가 있다는 문제점이 있죠.
2. 파일 객체를 한 줄씩 배열로 더 해주는 방법도 생각해 볼 수 있습니다.
with open('filename.txt', 'r') as f: list_file = [] for line in f: list_file.append(line)
append()가 속도가 빠른 것은 아니지만 이 방법이 1번보다 조금 더 효율적이고 안정적이라고 볼 수 있겠네요.
3. 심플하게 파일 객체를 '\n'으로 나눠서 각 요소를 리스트에 넣는 방법도 있습니다.
list_file = open('filename.txt', 'r').read().split('\n')
close()를 하지 않아도 대개의 경우 별다른 문제가 발생하지 않지만 모든 경우에 그런 것은 아니죠.
4. 다음과 같이 List Comprehension 한 줄로 처리하는 방법도 있습니다.
list_file = [line.rstrip('\n') for line in open('filename.txt', 'r')]
파이썬다운 문법이긴 한데 3번과 같은 문제점이 있기는 합니다.
5. '\n' 단위로 요소를 리스트에 넣되 '\n'을 ''로 바꿔주는 함수인 splitlines()를 이용할 수도 있습니다.
with open('filename.txt', 'r') as f: data = f.read() file_list = data.splitlines()
제가 보기에는 특별한 단점은 없는 것 같네요. 솔직히 어떤 문제점이 있는지는 잘 모르겠습니다 ㅠㅠ
원래 파이썬은 '가장 아름다운 하나의 답이 존재한다'라는 철학을 베이스로 하는데
저도 파이썬에 대해 잘 알지도 못하면서 작성하다보니 중구난방식의 글이 되어버렸네요 ㅠㅠ
스포어의 파이썬 고수님들께서 보시기엔 헛소리를 늘어놓는 것처럼 보일 수도 있겠네요.
그럼 굿밤 되세요~!