使用Python上传数据到数据库通常涉及以下步骤:
下载文件:
使用`requests`库下载文件,并将内容保存到本地。
转换数据:
根据数据库要求将文件数据转换为适当的格式,例如将CSV文件转换为字典或JSON格式。
连接数据库:
使用`pymysql`、`psycopg2`或其他数据库驱动程序与数据库建立连接。
准备SQL语句:
执行插入操作:
使用`execute()`方法执行SQL语句,并提供转换后的数据作为参数。
提交更改:
提交事务以将数据永久存储在数据库中。
import pandas as pd
import pymysql
读取Excel文件
filepath = 'E:\\_DataSet\\catering_sale.xls'
data = pd.read_excel(filepath)
建立数据库连接
db = pymysql.connect(host='localhost', user='root', passwd='数据库密码', db='python_analysis', charset='utf8')
获取游标对象
cursor = db.cursor()
创建数据库表(如果不存在)
try:
cursor.execute('CREATE TABLE IF NOT EXISTS catering_sale (num INT PRIMARY KEY, date DATETIME, sale FLOAT)')
except:
print('数据库已存在!')
准备插入语句
query = 'INSERT INTO catering_sale (num, date, sale) VALUES (%s, %s, %s)'
迭代读取每行数据并执行插入操作
for index, row in data.iterrows():
cursor.execute(query, (row['num'], row['date'], row['sale']))
提交事务
db.commit()
关闭数据库连接
db.close()
请确保替换`filepath`、`数据库密码`和数据库名称以匹配你的实际情况。如果你使用的是其他类型的数据库,比如PostgreSQL,你可能需要使用`psycopg2`库,并且连接参数也会有所不同。