1. 程式人生 > >向MySQL資料庫中批量讀取資料

向MySQL資料庫中批量讀取資料

這是在一次作業的時候用到的,將一個csv檔案中的資料讀取到MySQL建好的一個表中去。話不多說,上程式碼。
#coding=utf-8
import MySQLdb

conn = MySQLdb.connect(user='root',passwd='1122334455',db='Sales_Forecast_Qualification')
cur=conn.cursor()
f=open('t_sales_sum.csv','r')

while True:
    line = f.readline()
    if line:
        line = line.strip('\n')
        line = line.split(',')
        print line
        dt = line[0]
        shop_id = line[1]
        sale_amt_3m = line[2]
        cur.execute('insert into t_sales_sum(dt,shop_id,sale_amt_3m)values(%s,%s,%s)'%(dt,shop_id,sale_amt_3m))
    else:
        break

f.close()
cur.close()
conn.commit()
conn.close()

關於程式碼的解釋:

conn是一個連線資料庫的操作,括號內是相關引數,使用者名稱、密碼、連線的資料庫名稱等等。

cur是一個遊標,在一部分資料庫中需要使用的。

使用一個while迴圈,將csv檔案中的資料一行行讀取,並進行處理(主要是根據逗號把資料分開)。

cur.execute後面的括號是一個SQL的插入操作的語句。

操作結束後,關掉cur、commit、conn等。