使用python讀取csv文件快速插入數(shù)據(jù)庫(kù)的實(shí)例
如下所示:
# -*- coding:utf-8 -*- # auth:ckf # date:20170703 import pandas as pd import cStringIO import warnings from sqlalchemy import create_engine import sys reload(sys) sys.setdefaultencoding('utf8') warnings.filterwarnings('ignore') engine = create_engine( 'postgresql+psycopg2://'數(shù)據(jù)庫(kù)連接) 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 index df.to_csv(output, sep='\t',index = False, header = False) output.getvalue() # jump to start of stream output.seek(0) connection = engine.raw_connection() cursor = connection.cursor() # null value become '' cursor.copy_from(output,tablename,null='') connection.commit() cursor.close() print 'done!'
這個(gè)腳本可以直接運(yùn)行,將csv文件放在同級(jí)目錄即可。
csv第一列需要有列名,如果csv里沒(méi)有列名,需要在代碼中添加列名。
代碼運(yùn)行示例:python insert.py csvname tablename
以上這篇使用python讀取csv文件快速插入數(shù)據(jù)庫(kù)的實(shí)例就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
Django表單外鍵選項(xiàng)初始化的問(wèn)題及解決方法
這篇文章主要介紹了Django表單外鍵選項(xiàng)初始化的問(wèn)題及解決方法,需本文給大家介紹的非常詳細(xì),對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,要的朋友可以參考下2021-04-04python數(shù)據(jù)結(jié)構(gòu)之鏈表的實(shí)例講解
下面小編就為大家?guī)?lái)一篇python數(shù)據(jù)結(jié)構(gòu)之鏈表的實(shí)例講解。小編覺(jué)得挺不錯(cuò)的,現(xiàn)在就分享給大家,也給大家做個(gè)參考。一起跟隨小編過(guò)來(lái)看看吧2017-07-07python sklearn庫(kù)實(shí)現(xiàn)簡(jiǎn)單邏輯回歸的實(shí)例代碼
Scikit-learn(sklearn)是機(jī)器學(xué)習(xí)中常用的第三方模塊,對(duì)常用的機(jī)器學(xué)習(xí)方法進(jìn)行了封裝,這篇文章主要介紹了python sklearn庫(kù)實(shí)現(xiàn)簡(jiǎn)單邏輯回歸的實(shí)例代碼,需要的朋友可以參考下2019-07-07- 這篇文章主要為大家介紹了python中selenium模塊的安裝和配置環(huán)境變量教程、提取數(shù)據(jù)操作、無(wú)頭模式,有需要的朋友可以借鑒參考下,希望能夠?qū)Υ蠹矣兴鶐椭?/div> 2022-10-10
最新評(píng)論