这几天有个需求,简单说就是把一个2000x200的csv文件里的数据存放到MySQL里
网上有很多方法,但是导入的基本都是数据,Database里的每个field还是要自己手动创建,所以我大概要手动创建200个字段,以下是借助python的pymysql库写的创建字段脚本
import xlrd
import pymysql
data = xlrd.open_workbook(r'AV1SLtest.xlsx')
st = data.sheet_by_index(0)
col = st.ncols
fileds = [str(st.cell_value(0, i)) for i in range(1, col)]
# print(fileds[0])
mydb = pymysql.connect(
host="localhost",
user="root",
passwd="123456",
database="runoob3"
)
mycursor = mydb.cursor()
mycursor.execute("DROP TABLE IF EXISTS customers")
mycursor.execute("CREATE TABLE customers (name VARCHAR(255))")
for i in range(col - 1):
sql = "alter table customers add mts%s varchar(255)"%fileds[i]
mycursor.execute(sql)
mydb.commit()
mycursor.close()
mydb.close()
先将原先的csv文件存为xlxs,xlrd读取该文件,遍历读取所需字段,再调用pymysql里的方法创建字段