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!')
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值