将文件夹中的csv文件读入MySql数据库中的代码函数
时间: 2024-03-20 22:40:26 浏览: 80
以下是将指定文件夹中的所有csv文件读入MySql数据库中的Python代码函数:
```python
import os
import pandas as pd
from sqlalchemy import create_engine
def read_csv_to_mysql(folder_path, db_name, table_name, host='localhost', port=3306, user='root', password=''):
# 遍历文件夹中所有csv文件
csv_files = [os.path.join(folder_path, f) for f in os.listdir(folder_path) if f.endswith('.csv')]
if not csv_files:
print('No csv files found in the folder!')
return
# 创建MySql数据库连接
engine = create_engine(f'mysql+pymysql://{user}:{password}@{host}:{port}/{db_name}')
conn = engine.connect()
# 读入csv文件并写入数据库表
for csv_file in csv_files:
df = pd.read_csv(csv_file)
df.to_sql(table_name, conn, if_exists='append', index=False)
print(f'{csv_file} has been imported into {table_name} table in {db_name} database!')
conn.close()
```
其中,`folder_path`参数为指定文件夹路径,`db_name`参数为目标MySql数据库名称,`table_name`参数为目标表名称,`host`、`port`、`user`、`password`参数为MySql数据库连接参数。该函数会遍历指定文件夹中所有的csv文件,并将其依次读入到指定的MySql数据库表中。最后,函数会返回成功导入的csv文件数。
阅读全文