博客

如何将Pandas DataFrame写到PostgreSQL表中

  • Post category:Python

下面是将Pandas DataFrame写入PostgreSQL数据库的详细攻略，包括安装必要的库、数据库连接、数据表创建以及数据写入等步骤。

1. 安装必要的库

要将Pandas DataFrame写入PostgreSQL数据库，需要安装以下库：

  • pandas：用于数据处理和转换。
  • psycopg2：用于Python连接PostgreSQL数据库。

可以通过以下命令安装上述库：

pip install pandas psycopg2

2. 数据库连接

在连接PostgreSQL数据库之前，需要确保已经安装了PostgreSQL并且启动了服务。然后，可以通过以下代码连接PostgreSQL数据库：

import psycopg2

conn = psycopg2.connect(
    host="your_host_name",
    port=your_port_number,
    database="your_database_name",
    user="your_user_name",
    password="your_password"
)

其中，your_host_name是主机名，your_port_number是端口号，your_database_name是要连接的数据库名称，your_user_name是用户名，your_password是密码。

3. 数据表创建

在将DataFrame数据写入PostgreSQL数据库之前，需要创建一个相应的数据表。可以使用以下代码创建一个名为employees的数据表：

import pandas as pd
from psycopg2.extensions import AsIs

cur = conn.cursor()

cur.execute("""
    CREATE TABLE IF NOT EXISTS employees (
        id SERIAL PRIMARY KEY,
        name VARCHAR(100) NOT NULL,
        age INTEGER NOT NULL,
        salary NUMERIC(10,2) NOT NULL
    );
""")

conn.commit()

以上代码中，使用CREATE TABLE语句创建了一个名为employees的数据表，其中包括id、name、age和salary四个字段。其中PRIMARY KEY、NOT NULL以及NUMERIC(10,2)设置了数据表的约束条件。

4. 数据写入

在连接到PostgreSQL数据库并创建数据表后，可以将Pandas DataFrame 写入数据库中。可以通过以下代码将DataFrame中的数据写入到employees表中：

df = pd.DataFrame({
    'name': ['Alice', 'Bob', 'Charlie', 'David'],
    'age': [30, 40, 25, 35],
    'salary': [5000.00, 6000.50, 7500.10, 8000.00]
})

for _, row in df.iterrows():
    sql = "INSERT INTO employees (name, age, salary) VALUES (%s, %s, %s)"
    values = (row['name'], row['age'], row['salary'])
    cur.execute(AsIs(sql), values)

conn.commit()

以上代码中，首先创建了一个DataFrame对象，并将其存储在变量df中。然后使用df.iterrows()方法迭代DataFrame中的每一行。在接下来的循环中，将数据逐行写入到数据表中。

总结

在本文中，我们使用了Python语言和相关的库将Pandas DataFrame数据写入到PostgreSQL数据库中。如果按照上述4个步骤操作，就可以顺利将DataFrame数据写入到PostgreSQL数据库中。

Tags: