python和mysql交互详解---- (pandas)读csv文件,executemny批量写入db中

主要用到pandas从csv文件中抓数据,pandas抓出的数据是dataframe格式的,而且有的可能是Nan,抓出df格式的数据需要再处理,才能批处理的写入数据库中,executemany批出的格式不能是df,这里处理成list

import pymysql
import codecs
import pandas as pd
# 打开数据库连接
def get_conn():
    conn = pymysql.connect(host='xxxxx', port=3306, user='xx', passwd='xx', db='数据库名字', charset='utf8')
    return conn

def insert(cur, sql, args):
    cur.executemany(sql, args)

#从csv中读文件成数据
def read_csv(filename):
    #pandas读取csv获取dataframe格式的数据
    df = pd.read_csv(filename) 
    #将df中的Nan替换成none
    df = df.where(df.notnull(), None)
    #将df格式的数据转化为list格式
    dataList = df.values.tolist()
    return dataList

#将文件写入DB中        
def fileToDb(dataList,dbTableName):
    conn = get_conn()
    cur = conn.cursor()
    n = len(dataList[0])
    str_s = ','.join(['%s' for _ in range(n)])
    sql = 'insert into {}  values({})'.format(dbTableName,str_s)
      
    insert(cur, sql=sql, args=dataList)

    conn.commit()
    cur.close()
    conn.close()

调用函数

dataList = read_csv('orders.csv')
fileToDb(dataList,'ordersfinal')
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值