python 从csv读数据到mysql的例子

来源：互联网发布：初一数学教学软件编辑：程序博客网时间：2024/04/30 13:57

import csvimport sys,osimport MySQLdbdef 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 headersdef 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 curif __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()

0 0