Python如何连接PostgreSQL|极客教程

Python如何连接PostgreSQL

简介

PostgreSQL是一种功能齐全、强大的开源对象关系数据库管理系统。Python是一种流行的编程语言，拥有丰富的第三方库和框架。在本文中，我们将介绍如何使用Python连接到PostgreSQL数据库，并执行一些常见的操作。

Python中连接PostgreSQL的方式

Python中有多种方式可以连接到PostgreSQL数据库，其中比较常用的是使用第三方库psycopg2。psycopg2是一个为Python语言开发的PostgreSQL数据库适配器，可以帮助我们在Python中连接、操作PostgreSQL数据库。

安装`psycopg2`

在使用psycopg2之前，我们首先需要安装这个库。可以使用pip进行安装：

pip install psycopg2

连接到PostgreSQL数据库

连接到PostgreSQL数据库可以通过如下代码实现：

import psycopg2

# 连接到PostgreSQL数据库
conn = psycopg2.connect(
    dbname="your_database_name",
    user="your_username",
    password="your_password",
    host="your_host",
    port="your_port"
)

# 创建游标
cur = conn.cursor()

# 执行SQL查询
cur.execute("SELECT * FROM your_table")

# 读取查询结果
rows = cur.fetchall()
for row in rows:
    print(row)

# 关闭游标和连接
cur.close()
conn.close()

在上面的代码中，我们首先使用psycopg2.connect()方法连接到PostgreSQL数据库，指定数据库名称、用户名、密码、主机和端口。然后创建游标，执行SQL查询，读取查询结果并关闭游标和连接。

示例代码

接下来，我们给出一个完整的示例代码，演示如何连接到PostgreSQL数据库并执行一些常见的操作：

import psycopg2

# 连接到PostgreSQL数据库
conn = psycopg2.connect(
    dbname="postgres",
    user="postgres",
    password="password",
    host="localhost",
    port="5432"
)

# 创建游标
cur = conn.cursor()

# 创建表
cur.execute("""
    CREATE TABLE IF NOT EXISTS students (
        id SERIAL PRIMARY KEY,
        name VARCHAR(50) NOT NULL,
        age INT NOT NULL
    )
""")

# 插入数据
cur.execute("INSERT INTO students (name, age) VALUES ('Alice', 25)")
cur.execute("INSERT INTO students (name, age) VALUES ('Bob', 30)")

# 查询数据
cur.execute("SELECT * FROM students")
rows = cur.fetchall()
for row in rows:
    print(row)

# 关闭游标和连接
cur.close()
conn.close()

运行上面的代码，会连接到PostgreSQL数据库，创建一个名为students的表，插入两条数据，并查询所有数据并打印出来。