프로그래밍 문법/python

파이썬으로 파일 읽기, 쓰기

씩씩한 IT블로그 2020. 7. 5. 10:39
반응형

파일 읽기

1. 기본 내장 함수

    words = []
    f = open("words.txt", 'r')
    while True:
        line = f.readline().strip("\n")
        if not line: break
        words.append(line)
        '''
        코드처리
        '''
    f.close()

 

2. 기본 내장 함수 (with이용)

with open('경로/파일이름', 'r') as f:
    line = f.read()
    print(line)

 

3. 판다스(pandas) read_csv

csv파일을 읽을 때 사용

import pandas as pd

pd.read_csv("경로/파일이름")

 

 

파일 쓰기

1. 기본 내장 함수

f = open("경로/파일이름", 'w')
for i in range(10):
    line = f"{i}번째 줄"
    f.write(line)
f.close()

 

2. 기본 내장 함수(with 사용)

with open("경로/파일이름", "w") as f:
    f.write("이줄을 파일에 씁니다.")

 

3. 판다스(pandas) to_csv

import pandas as pd

df.to_csv('경로/파일이름',index={인덱스 없애고싶으면 False})

 

* to_csv를 사용하면 int타입의 값도 float로 바뀐다. 이를 방지하기 위해 다음의 코드를 추가 해준다.

df["column1"]=df["column1"].astype(int)
df["column2"]=df["column2"].astype(int)

df.to_csv('경로/파일이름')

 

반응형