[Python] 파일을 읽어서 각 행을 리스트로 만드는 5가지 방법

파이썬으로 스크립트를 작성하다보면 텍스트 파일을 읽어서 각 행을 리스트 변수에 넣어야 하는 경우가 종종 있죠.

 

이하 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()

 

제가 보기에는 특별한 단점은 없는 것 같네요. 솔직히 어떤 문제점이 있는지는 잘 모르겠습니다 ㅠㅠ

 

 

 

원래 파이썬은 '가장 아름다운 하나의 답이 존재한다'라는 철학을 베이스로 하는데 

 

저도 파이썬에 대해 잘 알지도 못하면서 작성하다보니 중구난방식의 글이 되어버렸네요 ㅠㅠ

 

스포어의 파이썬 고수님들께서 보시기엔 헛소리를 늘어놓는 것처럼 보일 수도 있겠네요.

 

그럼 굿밤 되세요~!