CSV 파일을 읽어보자!

승돌즈 프로그래밍

---

오늘의 주제

CSV 파일을 읽어야 할 일이 생겼습니다. 이유는 과제 도중 게임 AI log 파일이 100개 정도인데 하나 하나 열어 보기엔 너무 힘이 듭니다...심지어 Sum도 해야하기에 파이썬으로 프로그래밍을 해보았습니다. 사실 csv 파일도 아닙니다. 원래 형태는 다른 형태지만 csv 파일로 만드는 프로그램도 만들었습니다.^^추후에 다듬어서 py file로도 제공하겠습니다. 

CSV 파일 입출력 우선 기본기를 알아 보죠! 

>>> import csv
>>> of = file('test.csv','r')
>>> data = csv.reader(of)
>>> for item in data:
	    print item

	
['0', '648', '351']
['1', '500', '500']
['2', '500', '500']
>>> 

시작은 모듈을 불러오는 것으로 시작합니다. 

그 후 file을 오픈!! ㅎㅎ 

그리고 나서 csv.reader를 이용하여 객체 생성 후 data에 넣어줍니다. 

data를 순회하며, 해당 값들을 리스트 형태로 반환 받습니다. 

그런데!!! 난 정수가 필요해!!! 그렇다면, int(리스트[idx])를 해주시면 정수변환이 됩니다. 

그럼 이제 CSV 파일에 출력 해보아요!!

>>> out = file('blogout.csv','w')
>>> outWriter = csv.writer(out)
>>> od = []
>>> od.append(123)
>>> od.append(321)
>>> od
[123, 321]
>>> outWriter.writerow(od)
>>> 

쓰는 건 Writerow 함수를 통해 쓸 수 있습니다. 그리고 추가적으로 Writerows라는 함수도 존재합니다. 이건 써보지 않았네요 ;;;

결과적으로 데이터를 쓰고자 한다면, 리스트 생성 후 추가하신 후에 그 리스트 객체 자체를 Writerow 함수로 출력하게 되는 겁니다.