python 从csv读数据到mysql

代码语言：javascript复制

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,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

0 人点赞