python实现读取csv文件写入mysql

import pymysql
import pandas as pd

class Mysql_csv(object):

    # 定义一个init方法,用于读取数据库
    def __init__(self):
        # 读取数据库和建立游标对象
        self.connect = pymysql.connect(host="192.168.146.133", \
            port=3306, user="root", password="root", \
                database="superset", charset="utf8")
        self.cursor = self.connect.cursor()
        
    # 定义一个del类,用于运行完所有程序以后关闭数据源和游标对象
    def __del__(self):
        self.connect.close()
        self.cursor.close()
        
    # 读取csv文件的列索引,用来建立数据表的字段
    def read_csv_colnmns(self):
        # 读取csv文件的索引
        # csv_name='E:\\电网数据\\新能源\\2017山西.csv'
        csv_name='C:\\Users\\23892\\Desktop\\job.csv'
        data = pd.read_csv(csv_name, encoding="utf-8")
        data_list = data.values
        return data_list
        
    # 往数据库写入数据
    def write_mysql(self):
        #在数据表中写入数据,因为数据是列表类型,把他转化为元组更符合sql语句
        for i in self.read_csv_colnmns():
            data = tuple(i)
            print(data)
            sql = "insert into job values{}".format(data)
            self.cursor.execute(sql)
            self.commit()
        print("\n数据写入完成")
        
    # 定义一个确认事务运行的方法
    def commit(self):
        self.connect.commit()
        
    # 新建数据库表
    def create(self):
        # 若已有数据库表shanxi,则删除
        query = "drop table if exists job;" # 更改表名
        self.cursor.execute(query)
        # 创建数据表,用刚才提取的列索引作为字段
        data_2 = self.read_csv_colnmns()
        # 根据自己要创建的表格更改sql语句
        sql = "create table if not exists job(Position varchar(50) not null,Level int(50) not null, Salary int(50) not null, primary key(Position))default charset=utf8;"
        self.cursor.execute(sql)
        self.commit()

    # 运行程序
    def run(self):
        self.create()
        self.write_mysql()
    

# 封装函数
def main():
    sql = Mysql_csv()
    sql.run()
if __name__ == '__main__':
    main()

  • 0
    点赞
  • 20
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值