将DataFrame某列插入到mysql表的指定列
时间: 2023-03-29 19:04:08 浏览: 371
spark rdd转dataframe 写入mysql的实例讲解
可以使用 pandas 库的 to_sql 方法将 DataFrame 直接写入 MySQL 数据库中,示例代码如下:
```python
import pandas as pd
from sqlalchemy import create_engine
# 创建 MySQL 数据库连接
engine = create_engine('mysql+pymysql://username:password@host:port/database')
# 读取 DataFrame 数据
df = pd.read_csv('data.csv')
# 将 DataFrame 写入 MySQL 数据库中
df.to_sql(name='table_name', con=engine, if_exists='append', index=False)
```
其中,name 参数为表名,con 参数为数据库连接,if_exists 参数为数据存在时的处理方式,index 参数为是否将 DataFrame 的索引写入数据库中。如果需要将 DataFrame 的某一列插入到 MySQL 表的指定列中,可以先将 DataFrame 转换为字典,再使用 MySQL 的 INSERT INTO 语句插入数据,示例代码如下:
```python
import pandas as pd
import pymysql
# 创建 MySQL 数据库连接
conn = pymysql.connect(host='host', port=port, user='username', password='password', database='database')
# 读取 DataFrame 数据
df = pd.read_csv('data.csv')
# 将 DataFrame 转换为字典
data = df.to_dict(orient='records')
# 插入数据到 MySQL 表中
cursor = conn.cursor()
for row in data:
sql = "INSERT INTO table_name (col1, col2, col3) VALUES (%s, %s, %s)"
cursor.execute(sql, (row['col1'], row['col2'], row['col3']))
conn.commit()
cursor.close()
conn.close()
```
其中,col1、col2、col3 分别为 MySQL 表中的列名,可以根据实际情况进行修改。
阅读全文