파일에서 읽고 쓰기

외부의 어떤 데이터를 읽어와서 작업하고 다시 파일에 쓰는 작업은 기본적이지만 파이썬을 처음 하는 입장에서는 생소한 것이 사실이다. 파이썬으로 데이터를 읽고 쓰는 것을 해 보겠다. 

예제파일은 주식 종목과 현재 가격이 있는 파일이고 종목명과 현재가를 나눠서 두개의 파일로 저장해 보겠다. 

1. os 모듈 임포트 하고 작업파일 불러오기 

import os
os.chdir('c:\\temp')

 

2. 파일을 읽어오기

stocks = open('test1.csv')

 

3. 파일을 한 줄씩 읽어오고 각 항목을 그룹으로 나누기 
agroup = []
bgroup = []

for i in stocks:
(name, price) = i.split(',')
agroup.append(name)
bgroup.append(price)

 

4. 파일쓰기_csv와 txt파일로 저장가능

with open('a.csv','w') as agroup_csv:
print(agroup, end='', file=agroup_csv)

with open('b.txt','w') as bgroup_txt:
print(bgroup, file=bgroup_txt)