帮我写一段python代码 用于 读取桌面qaz目录下的以_t0结尾的excel文件内容 写入至mysql(要求1: 将表头"日期" 改为 "transaction_date",将表头"市场" 改为 "trade_type",将表头"交易金额" 改为"amount" ; 要求2: 将字符集编码改为utf-8 且中文字符集不能有乱码)
时间: 2023-07-05 10:19:46 浏览: 104
好的,这里是一份可能的代码供您参考:
```python
import os
import pandas as pd
import pymysql
# 连接 MySQL 数据库
conn = pymysql.connect(host='localhost', user='root', password='your_password', database='your_database', charset='utf8mb4')
# 获取桌面 qaz 目录下所有以 _t0 结尾的 Excel 文件路径
file_paths = [os.path.join(os.path.expanduser('~'), 'Desktop', 'qaz', f) for f in os.listdir(os.path.join(os.path.expanduser('~'), 'Desktop', 'qaz')) if f.endswith('_t0.xlsx')]
for file_path in file_paths:
# 读取 Excel 文件内容
df = pd.read_excel(file_path)
# 修改表头列名
df = df.rename(columns={'日期': 'transaction_date', '市场': 'trade_type', '交易金额': 'amount'})
# 将数据写入 MySQL 数据库
df.to_sql(name='table_name', con=conn, if_exists='append', index=False)
# 关闭数据库连接
conn.close()
```
请将 `your_password` 和 `your_database` 替换为实际的 MySQL 连接密码和数据库名,将 `table_name` 替换为您要写入数据的实际表名。此外,您还需要安装 `pandas` 和 `pymysql` 库。
阅读全文