帮我写一段python代码 用于 读取D盘qaz目录下的以_t0结尾的文件内容 并根据表头字段将内容将其写入至mysql对应表字段中(要求1: 将表头"日期" 改名为 "transaction_date",将表头"市场" 改名为 "trade_type",将表头"交易金额" 改名为"amount" ; 要求2: 将 encoding='gbk' 转为 encoding='utf-8' 中文字符集不要乱码)
时间: 2023-07-06 22:22:27 浏览: 90
好的,以下是代码示例:
```python
import os
import pandas as pd
import pymysql
# 连接 MySQL 数据库
conn = pymysql.connect(host='localhost', user='root', password='password', database='test', charset='utf8mb4')
# 读取 D 盘 qaz 目录下以 _t0 结尾的文件
file_list = [f for f in os.listdir('D:\\qaz') if f.endswith('_t0')]
for file_name in file_list:
# 读取文件内容
df = pd.read_csv(f'D:\\qaz\\{file_name}', encoding='gbk')
# 修改表头字段名
df = df.rename(columns={'日期': 'transaction_date', '市场': 'trade_type', '交易金额': 'amount'})
# 将数据写入 MySQL 数据库
df.to_sql('table_name', conn, if_exists='append', index=False)
# 关闭数据库连接
conn.close()
```
注意,这段代码需要修改的地方有:
1. 数据库连接信息:需要根据实际情况修改 host、user、password、database 等参数。
2. 表名:需要将 `table_name` 修改为实际使用的表名。
3. 字符集:需要根据实际情况修改原始文件的字符集,本例中使用的是 `gbk`,如果原始文件是其他字符集需要相应修改。
4. 文件路径:需要将 `D:\\qaz` 修改为实际文件所在的路径。
希望对你有帮助!
阅读全文