如下所示:
# -*- coding:utf-8 -*-# auth:ckf# date:20170703import pandas as pdimport cStringIOimport warningsfrom sqlalchemy import create_engineimport sysreload(sys)sys.setdefaultencoding('utf8')warnings.filterwarnings('ignore')engine = create_engine( 'postgresql+psycopg2://'數據庫連接)filename = sys.argv[1]tablename = sys.argv[2]print '=== csvname is',filename ,'tablename is',tablename,'==='print 'read', filename, '...'df = pd.read_csv(filename, sep=';')print 'read', filename, 'done!'print 'lets insert ...'output = cStringIO.StringIO()# ignore the indexdf.to_csv(output, sep='/t',index = False, header = False)output.getvalue()# jump to start of streamoutput.seek(0)connection = engine.raw_connection()cursor = connection.cursor()# null value become ''cursor.copy_from(output,tablename,null='')connection.commit()cursor.close()print 'done!'
這個腳本可以直接運行,將csv文件放在同級目錄即可。
csv第一列需要有列名,如果csv里沒有列名,需要在代碼中添加列名。
代碼運行示例:python insert.py csvname tablename
以上這篇使用python讀取csv文件快速插入數據庫的實例就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持武林站長站。
新聞熱點
疑難解答