向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等。

你可能感兴趣的:(向MySQL数据库中批量读取数据)