帮我写一段python代码 用于 读取桌面qaz目录下的以_t0结尾的excel文件内容 写入至mysql(要求: 将表头“日期” 改为 “transaction_date”,将表头"市场" 改为 “trade_type”,将表头“交易金额” 改为“amount” )
时间: 2023-09-21 10:06:26 浏览: 95
python读取excel内容上传到mysql
好的,以下是一份满足您要求的 Python 代码:
```python
import os
import pandas as pd
import pymysql
# 连接 MySQL 数据库
conn = pymysql.connect(host='localhost', user='root', password='password', db='test', charset='utf8mb4')
# 读取桌面 qaz 目录下的所有 _t0 结尾的 Excel 文件
dir_path = os.path.expanduser('~/Desktop/qaz')
for file_name in os.listdir(dir_path):
if file_name.endswith('_t0.xlsx'):
file_path = os.path.join(dir_path, file_name)
# 读取 Excel 文件内容
df = pd.read_excel(file_path)
# 修改表头
df = df.rename(columns={'日期': 'transaction_date', '市场': 'trade_type', '交易金额': 'amount'})
# 将数据写入 MySQL 数据库
df.to_sql('table_name', conn, if_exists='append', index=False)
# 关闭数据库连接
conn.close()
```
在以上代码中,我们使用了 pandas 来读取 Excel 文件内容,并使用 pymysql 将数据写入 MySQL 数据库。其中,`os.listdir` 可以遍历一个目录下的所有文件,`df.rename` 可以修改表头,`df.to_sql` 可以将数据写入 MySQL 数据库。请注意,您需要根据实际情况修改代码中的数据库连接信息和表名。
阅读全文