[python] csv 파일 읽기, 쓰기

CSV(comma seperated  values, 쉼표로 구분된 값들)파일을 읽어서 처리하는 코드를 작성하고보니 업로드 해야겠다는 생각이 들었다.

python 에서는 csv라는 모듈을 제공해준다. 그래서 import csv를 해서 사용하면 된다.

- csv 파일 읽기

import csv
 
def getCsv(self):
        csvList = []
    
        f = open('./exam.csv','r')    #현재 경로의 exam.csv를 연다.
        csvReader = csv.reader(f)    #reader로 파일을 읽는다.
        
        for i in csvReader:            #한 행씩 돌면서 i[2]값 (3번째 컬럼)을 가져와서 리스트에 저장한다.
            csvList.append(i[2])
        f.close()
    
        return csvList
        
def checker(self):
    csvList = self.getCsv()
    
    if 'ecycle' not in csvList:        #리스트에서 ecycle 문구 존재여부 체크
        print "X"
    else:
        print "O"    
 
if __name__ =='__main__':
    
    print sta.Checker()    

- csv 파일 쓰기

def setCsv(self):
    
    f = open('./exma2.csv','w')
    csvWriter = csv.writer(f)
        
    csvWriter.writerow(['ecycle','1989','seoul','preprofession'])
    f.close()