PostgreSQL使用单个脚本分两个阶段导入CSV文件(psycopg2)

import psycopg2
import csv

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

# 创建表格
cur.execute("CREATE TABLE IF NOT EXISTS your_table (column1 datatype1, column2 datatype2, ...)")
conn.commit()

# 定义CSV文件路径和表格名称
csv_file = "path_to_csv_file.csv"
table_name = "your_table"

# 打开CSV文件并读取数据
with open(csv_file, 'r') as file:
    reader = csv.reader(file)
    next(reader)  # 跳过标题行
    for row in reader:
        # 插入数据到表格中
        cur.execute("INSERT INTO {} VALUES (%s, %s, ...)".format(table_name), row)
    conn.commit()

# 关闭数据库连接
cur.close()
conn.close()