1. 程式人生 > >python查詢資料庫並導處csv檔案

python查詢資料庫並導處csv檔案

# -*- coding: utf-8 -*-

import MySQLdb
import math
import csv
import codecs
import sys
reload(sys)
sys.setdefaultencoding('utf-8')

csvfile = open('sample.csv', 'wb')
csvfile.write(codecs.BOM_UTF8)     # 防止中文亂碼!!!
writer = csv.writer(csvfile,dialect='excel')
#開始寫第一行  陣列的形式
writer.writerow(['id', 'entry'
, 'type']) try: connect = MySQLdb.connect(host='localhost', user='root', passwd='', db='test', charset='utf8') db = connect.cursor() db.execute("select * from words where annotator="+"'"+annotator+"'") #忽略查詢語句 data=db.fetchall() #寫入多行格式 ''' data=[ (1,2,3), (4,5,6), (7,8,9) ] '''
writer.writerows(data) writer.writerows([(),(),()]) #三個空白行 csvfile.close() connect.commit() db.close() connect.close() print "Done!" except MySQLdb.Error,e: print "MySQL Error %d:%s"%(e.args[0],e.args[1])

讀csv檔案程式碼:

# -*- coding: utf-8 -*-
import csv
import sys
reload(sys)
sys.setdefaultencoding('utf-8'
) csv_reader = csv.reader(open('test.csv', 'rb')) for row in csv_reader: data={} #自己根據打印出來的資料操作~~