1. 程式人生 > 資料庫 >python 從csv讀資料到mysql的例項

python 從csv讀資料到mysql的例項

如下所示:

import csv
import sys,os
import MySQLdb

def read_csv(filename):
  with open(filename) as f:
    f_csv = csv.reader(f)
    headers = next(f_csv)
    #資料格式[1111,22222,1111,.....]
    #for row in f_csv:
      # Process row
    #  field1=row[1]
    #  ...
    return headers

def conn_to_psto():
  #mysql連線方法
  #MySQLdb.connect()
  #postgl連線
  onn = MySQLdb.connect(database="testdb",\
              user="postgres",\
              password="pass123",\
              host="127.0.0.1",\
               port="5432")
  cur = conn.cursor()
  return cur

if __name__ == "__main__":
  #傳入檔案路徑或檔名
  filename=sys.argv[1]
  f_csv=read_csv(filename)
  cur=conn_to_psto()
  for row in f_csv:
      # Process row
      field1=row[1]
      ...
    sql='''insert into .......(field1,field2,...) values(%,%,...) '''.format(field1=field1,field2=field2.........))
    cur.execute(sql)
  conn.commit()
  conn.close()

以上這篇python 從csv讀資料到mysql的例項就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。