飞道的博客

Python连接MySQL将csv文件存入数据库

369人阅读  评论(0)

此处使用 Python3 和 MySQL数据库连接,并读取csv文件,写入数据库。

这里我们将使用 PyMySQL库实现 Python 和 MySQL数据库交互。

PyMySQL库安装方法:在命令提示符中输入pip3 install PyMySQL


此处我准备将之前爬取的豆瓣电影Top100.csv文件存入数据库,打开文件预览:
第一列是自动生成的序号,不需要将它入库;第一行我们可以使用 SQLyog 设置,因此也不需要入库。

这里我们事先使用 SQLyog 创建 DataBase:dbmovie_top100 和 Table:db_top100:


并把 Table各列的参数配置好。(例如说我们需要的第一列是排名,设置数据类型为int)


接下来使用代码实现入库:

import csv
import pymysql

# 连接MySQL数据库(注意:charset参数是utf8而不是utf-8)
conn = pymysql.connect(host='localhost', user='root', password='MySQL密码', db='dbmovie_top100', charset='utf8')

# 创建游标对象
cursor = conn.cursor()

# 读取csv文件
with open('豆瓣电影Top100.csv', 'r', encoding='utf-8') as f:
    read = csv.reader(f)
    
    # 一行一行地存,除去第一行和第一列
    for each in list(read)[1:]:
        i = tuple(each[1:])        
        # 使用SQL语句添加数据
        sql = "INSERT INTO db_top100 VALUES" + str(i) # db_top100是表的名称
        cursor.execute(sql) #执行SQL语句
    
    conn.commit() # 提交数据
    cursor.close() # 关闭游标
    conn.close() # 关闭数据库

运行程序:

成功将csv文件存入MySQL数据库。


转载:https://blog.csdn.net/weixin_45961774/article/details/105099974
查看评论
* 以上用户言论只代表其个人观点,不代表本网站的观点或立场