# - * -编码:utf - 8 - *
#身份验证:位置
日期:20170703
熊猫作为pd导入
进口cStringIO
进口警告
从sqlalchemy进口create_engine
导入系统
重载(系统)
sys.setdefaultencoding (use utf8)
warnings.filterwarnings(“忽略”)
引擎=create_engine (
postgresql + psycopg2://的数据库连接)
文件名=sys.argv [1]
表名=sys.argv [2]
打印“===csvname”,文件名,表名是,表名,“===?
打印“读”,文件名,“……”
df=pd。read_csv(文件名,9=';')
打印“读”,文件名,“完成了!”
打印“允许插入…”
输出=cStringIO.StringIO ()
#忽略指数
df。to_csv(输出,9=\ t,指数=False,头=False)
output.getvalue ()
#跳转到流的开始
output.seek (0)
连接=engine.raw_connection ()
光标=connection.cursor ()
#空值成为“
cursor.copy_from(输出,表名,null=")
connection.commit ()
cursor.close ()
打印“做!”
之前
这个脚本可以直接运行,将csv文件放在同级目录即可。
csv第一列需要有列名,如果csv里没有列名,需要在代码中添加列名。
以上这篇使用python读取csv文件快速插入数据库的实例就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。
