파이썬 파일 입출력 기초 사용법

파이썬 파일 입출력 기초 사용법

코딩을 시작하면서 데이터 관리의 중요성을 느끼는 경우가 많습니다. 특히 파이썬(Python)에서는 파일 입출력이 필수적인 작업 중 하나로, 데이터를 저장하고 불러오는 방법을 알고 있으면 여러모로 유용합니다. 이번 글에서는 파이썬에서 파일을 생성하고, 데이터를 기록하며, 읽어오는 기본적인 방법을 안내해드리겠습니다.

파일 생성 및 데이터 기록하기

파이썬에서 파일을 만들고 데이터를 기록하는 가장 기본적인 방법은 open() 함수를 사용하는 것입니다. 이 함수는 두 개의 주요 인수를 요구합니다: 파일 경로와 모드입니다. 모드에는 'w' (쓰기), 'a' (이어쓰기), 'r' (읽기) 등이 있습니다.

예를 들어, 아래와 같은 코드를 통해 새로운 텍스트 파일을 생성하고 데이터를 기록할 수 있습니다:

file_path = "output.txt"
with open(file_path, "w") as file:
  file.write("Hello, World!\n")
  file.write("이것은 새로운 줄입니다.")

위 코드에서는 “output.txt”라는 이름으로 파일을 생성하고, 두 줄의 메시지를 기록합니다. with 문을 이용하면, 작업이 완료된 후 파일이 자동으로 닫혀 관리가 용이합니다.

이어쓰기 기능 사용하기

기존 파일에 새로운 내용을 추가하고 싶다면 'a' 모드를 사용합니다. 아래 코드를 참고해주세요:

with open(file_path, "a") as file:
  file.write("\n이 줄은 추가된 내용입니다.")

이렇게 하면 기존 내용 아래에 새로운 내용이 이어서 기록됩니다.

파일 경로 지정하기

파일을 지정할 때 경로를 잘 설정하는 것이 중요합니다. 특히 하위 폴더에 저장하고자 한다면, 경로를 정확히 입력해야 합니다. 예시로는 다음과 같은 방식이 있습니다:

open('folder/output.txt', 'w')

이 코드는 “folder”라는 하위 폴더에 “output.txt”라는 파일을 생성합니다. 경로에 공백이 포함될 경우, 따옴표로 감싸거나 탈출 문자를 사용해야 합니다.

명령 프롬프트를 통한 파일 저장

또한, 파이썬 코드를 명령 프롬프트에서 직접 실행하고 그 결과를 파일로 저장할 수도 있습니다. 이를 위해 아래와 같은 명령어를 사용할 수 있습니다:

python test.py > output.txt

이 명령어는 실행 결과를 “output.txt”라는 파일에 기록합니다. 이어쓰기를 원할 경우 >> 기호를 사용하는 것을 잊지 마세요.

glob 모듈 활용하기

파이썬의 glob 모듈은 파일 이름 패턴에 따라 파일과 폴더를 검색하는 데 유용합니다. 특정 패턴을 가진 파일을 쉽게 찾아낼 수 있는 방법을 살펴보겠습니다.

현재 디렉토리 내 파일과 폴더 탐색하기

• 현재 디렉토리의 모든 폴더를 확인하려면, 아래와 같이 사용합니다:

for folder_name in glob.glob('*/'):
  print(folder_name)

• 파일과 폴더를 모두 확인하려면, 아래의 코드를 사용합니다:

for f in glob.glob('*'):
  print(f)

특정 확장자 파일 찾기

특정 확장자의 파일을 찾고자 할 때는 아래와 같은 사용법이 유용합니다:

for file_name in glob.glob('*.txt'):
  print(file_name)

이 코드는 현재 디렉토리에서 “.txt” 확장자를 가진 모든 파일을 출력합니다. 하위 폴더를 포함하고 싶을 경우, recursive=True를 설정하여 검색할 수 있습니다.

데이터 핸들링: CSV 파일 저장하기

데이터 분석 및 관리 시 CSV 파일은 매우 흔히 사용되는 형식입니다. 파이썬에서는 pandas 라이브러리를 통해 쉽게 CSV 파일을 생성하고 저장할 수 있습니다.

CSV 파일 저장하기

먼저 pandas 라이브러리를 임포트하고 데이터 프레임을 생성한 후, CSV 파일로 저장하는 방법을 살펴보겠습니다:

import pandas as pd
data = {
  'Name': ['Alice', 'Bob'],
  'Score': [90, 85]
}
df = pd.DataFrame(data)
df.to_csv('scores.csv', index=False)

이 코드는 “scores.csv”라는 파일로 데이터 프레임을 저장하게 됩니다. index=False 옵션을 통해 인덱스 열을 저장하지 않을 수 있습니다.

데이터 불러오기

저장된 CSV 파일을 불러와 데이터 프레임으로 변환하는 것도 매우 간단합니다:

df_loaded = pd.read_csv('scores.csv')

이제 df_loaded 변수에 CSV 파일의 내용이 저장되고, 이를 쉽게 사용할 수 있습니다.

결론

이상으로 파이썬을 이용한 파일 입출력 기본 사용법에 대해 살펴보았습니다. 파일을 생성하고 데이터를 기록하거나 불러오는 방법, 그리고 glob 모듈을 활용하여 파일을 탐색하는 방법, CSV 파일을 처리하는 방법까지 다양한 내용을 다루어 보았습니다. 이러한 기초적인 작업을 숙지해 두면, 데이터 관리 및 분석에 훨씬 유리할 것입니다.

더 나아가, 파이썬의 파일 입출력 관련 기능을 마스터하면 데이터 분석, 머신러닝 등의 분야에서도 유용하게 사용될 수 있습니다. 이제 여러분도 파이썬을 활용하여 다양한 데이터를 손쉽게 관리해 보시기 바랍니다.

자주 물으시는 질문