如何使用Python将CSV文件导入到数据库

CSV（Comma Separated Values）是一种可读性高、易于编辑与导入导出的文件格式，常用于存储表格数据。在数据处理过程中，我们有时需要将CSV文件导入到数据库中进行进一步处理和分析。Python是一种强大的编程语言，可以轻松实现CSV文件到数据库的转换。本文将介绍Python如何导入CSV文件到数据库。

一、连接到数据库

在将CSV文件导入到数据库前，需要先连接到数据库。

import psycopg2

conn = psycopg2.connect(database="mydb", user="myuser", password="mypassword", host="localhost", port="5432")
cur = conn.cursor()

二、读取CSV文件

Python提供了csv模块，可以方便地读取CSV文件中的数据。

import csv

with open('data.csv', 'r') as csvfile:
    csvreader = csv.reader(csvfile)
    for row in csvreader:
        print(row)

三、创建表

在将CSV文件中的数据导入到数据库之前，需要先创建数据库表。

cur.execute("CREATE TABLE mytable (id INT PRIMARY KEY NOT NULL, name TEXT, age INT)")
conn.commit()

四、插入数据

将CSV文件中的数据插入到数据库表中。

with open('data.csv', 'r') as csvfile:
    csvreader = csv.reader(csvfile)
    next(csvreader) # skip header row
    for row in csvreader:
        cur.execute("INSERT INTO mytable (id, name, age) VALUES (%s, %s, %s)", row)
    conn.commit()

五、关闭连接

数据导入完成后，不要忘记关闭数据库连接。

cur.close()
conn.close()

六、完整代码

import csv
import psycopg2

conn = psycopg2.connect(database="mydb", user="myuser", password="mypassword", host="localhost", port="5432")
cur = conn.cursor()
cur.execute("CREATE TABLE mytable (id INT PRIMARY KEY NOT NULL, name TEXT, age INT)")
with open('data.csv', 'r') as csvfile:
    csvreader = csv.reader(csvfile)
    next(csvreader) # skip header row
    for row in csvreader:
        cur.execute("INSERT INTO mytable (id, name, age) VALUES (%s, %s, %s)", row)
    conn.commit()
cur.close()
conn.close()

七、总结

通过Python和psycopg2库的组合，我们可以将CSV文件中的数据快速地导入到数据库中。同时，我们还学习了如何连接到数据库、读取CSV文件、创建表和插入数据等操作。以上方法不仅适用于PostgreSQL数据库，也适用于其他常用的关系型数据库，如MySQL和Oracle等。