python:csv写入mysql

import pandas as pd
import pymysql

def company_to_db(data_path, db_params):
    data = pd.read_csv(data_path, header=None, engine='python')
    db = pymysql.connect(db_params['host'], db_params['username'], db_params['password'], db_params['db_name'])
    cursor = db.cursor()
    try:
        sql = 'drop table if exists company_csv'
        cursor.execute(sql)
        sql = 'create table company_csv(' \
              ......) default charset=utf8'
        cursor.execute(sql)
        sql = 'insert into company_csv values("'+row[0]+'","'+row[1]+'","'+row[2]+'","'+row[3]+'","'+row[4]+'","'+row[5]+'")'
        cursor.execute(sql)
        db.commit()
        return True
    except:
        db.rollback()
        return False
    finally:
        db.close()

if __name__ == '__main__':
    company_csv = 'D:/5-Python/爬虫/data/company.csv'
    db_params = {
        'host' : '127.0.0.1',
        'username' : 'root',
        'password' : '',
        'db_name' : 'test'
    }

    company_ret = company_to_db(company_csv, db_params)
    if company_ret:
        print ('company csv to db succeed!')
    else:
        print ('company csv to db failed, try again!')

你可能感兴趣的:(数据领域)