python从csv文件中读取数据添加到数据库中

最新推荐文章于 2024-11-25 13:58:30 发布

原创最新推荐文章于 2024-11-25 13:58:30 发布 · 4k 阅读

6 ·

CC 4.0 BY-SA版权

python 专栏收录该内容

82 篇文章

订阅专栏

本文介绍如何使用Python的csv模块与SQLAlchemy ORM将CSV文件中的数据批量导入到PostgreSQL数据库中。示例代码展示了从CSV文件读取数据，并通过SQLAlchemy执行SQL语句完成数据插入的过程。

示例如下，

其中csv是python内置的一个模块，专门用来处理csv文件的读写，不需要再另外下载。

:origin和:destination以及:duration都是占位符，后面的字典表示用o取代origin占位符来执行sql语句

import csv

from sqlalchemy import create_engine
from sqlalchemy.orm import scoped_session, sessionmaker

engine = create_engine('postgresql+psycopg2://postgres:88595073@localhost/lecture3')
db = scoped_session(sessionmaker(bind=engine))

def main():
    f = open("flights.csv")
    reader = csv.reader(f)
    for o, dest, dur in reader:
        db.execute("INSERT INTO flights (origin, destination, duration) VALUES (:origin, :destination, :duration)",
                    {"origin": o, "destination": dest, "duration": dur})
        print(f"Added flight from {o} to {dest} lasting {dur} minutes.")
    db.commit()

if __name__ == "__main__":
    main()