Python 将DataFrame写入数据库表|极客教程

Python 将DataFrame写入数据库表

在数据处理和分析过程中，经常会使用到Pandas库来处理数据。而有时为了将处理好的数据保存到数据库中，需要将Pandas的DataFrame对象写入数据库表中。在Python中，可以使用to_sql方法来实现这一功能。

使用`to_sql`方法写入数据库表

to_sql方法是Pandas库中DataFrame对象的方法，用于将DataFrame对象写入数据库表。这个方法的基本语法如下：

DataFrame.to_sql(name, con, schema=None, if_exists='fail', index=True, index_label=None, chunksize=None, dtype=None, method=None)

参数说明：

name：要写入的数据库表的表名
con：数据库连接对象
schema：数据库模式（可选）
if_exists：写入方式，有三种可选值：’fail’、’replace’、’append’，默认为’fail’
index：是否将DataFrame的索引写入数据库，默认为True
index_label：指定索引名称（可选）
chunksize：一次提交的数据块大小，默认为None
dtype：列的数据类型（可选）
method：指定写入方式（可选）

接下来，我们通过示例代码来演示如何使用to_sql方法将DataFrame写入数据库表。

示例一：将DataFrame写入MySQL数据库表

首先，我们需要安装所需的库：

pip install pymysql sqlalchemy

然后，我们创建一个DataFrame对象：

import pandas as pd

data = {
    'id': [1, 2, 3],
    'name': ['Alice', 'Bob', 'Charlie'],
    'age': [25, 30, 35]
}

df = pd.DataFrame(data)

print(df)

运行结果如下：

   id     name  age
0   1    Alice   25
1   2      Bob   30
2   3  Charlie   35

接着，我们连接到MySQL数据库，并将DataFrame写入一个名为users的表中：

from sqlalchemy import create_engine

engine = create_engine('mysql+pymysql://username:password@localhost/db_name')
df.to_sql('users', con=engine, if_exists='replace', index=False)

这样就将DataFrame成功写入了MySQL数据库表中。

示例二：将DataFrame写入SQLite数据库表

同样，我们首先安装所需的库：

pip install sqlalchemy

然后，我们可以创建一个DataFrame对象：

import pandas as pd

data = {
    'id': [1, 2, 3],
    'content': ['Python is powerful', 'Pandas is great', 'Geek-docs.com'],
    'category': ['Programming', 'Data Analysis', 'Documentation']
}

df = pd.DataFrame(data)

print(df)

运行结果如下：

   id               content      category
0   1     Python is powerful   Programming
1   2        Pandas is great   Data Analysis
2   3          Geek-docs.com   Documentation

接着，我们连接到SQLite数据库，并将DataFrame写入一个名为articles的表中：

from sqlalchemy import create_engine

engine = create_engine('sqlite:///mydatabase.db')
df.to_sql('articles', con=engine, if_exists='replace', index=False)