这是在一次作业的时候用到的,将一个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等。