MySQL 是目前最为流行的关系型数据库。接下来我们通过一个简单的案例,练习如何使用 PyMySQL 模块进行增删改查的操作。简单来讲,核心步骤有两步:链接数据库,读取 CSV 文件并导入数据表,然后查询数据,将数据表和计算结果导出为 CSV 文件。下面一起来具体看看吧~
Step1: 安装 PyMySQL 模块
在正式动手之前,我们需要先安装 PyMySQL 模块。
(1)使用 pip 安装, 清华镜像:
pip install -i https://pypi.tuna.tsinghua.edu.cn/simple pymysql
(2)使用 conda 安装
conda install pymysql
Step2: 从 CSV 文件导入数据到 MySQL 数据表
安装好以后,我们来进行第二步操作,从 CSV 文件导入数据到 MySQL 数据表。与内置的 SQLite 模块一样, PyMySQL 也遵循 DB-API 规范,因此我们前面开发的大多数代码经过简单修改即可使用,以下代码中主要注释了与 SQLite 操作的不同之处。
# 导入 pymysql 模块
import pymysql
# 连接数据库, 这里需要提供许多参数给 connect 方法
# host 服务器ip地址
# user 用户名 password 密码
# db 数据库名称
cn = pymysql.connect(host='localhost', user='pandas', password='pandas', db='pandas')
# port 端口 缺省值为 3306
# 新建游标对象 :和 SQLite 一样
cur = cn.cursor()
# 建表SQL语句, 小技巧: if not exists 不会引起表已经存在的错误
# MySQL 使用的是 auto_increment , 注意有个下划线
artist_create_table = """
create table if not exists artist(
id integer primary key auto_increment not null,
name varchar(255)
)
"""
try:
cur.execute(artist_create_table)
print(" 成功建表 ")
except:
pass
# 插入语句 , 使用 %(id)s 意味着我们会用字典对象替换数据
# PyMYSQL 使用的语法是 %(名称)s
artist_insert ="""
insert into artist(id, name) values ( %(id)s, %(name)s )
"""
# 以下代码和 SQLite 一样
# 导入 csv 模块
import csv
# 用 with 语法, 打开文件
with open('artist.csv', newline='') as csvfile:
# 用 DictReader 类, 方便开发
reader = csv.DictReader(csvfile)
# 按行遍历csv 文件
for row in reader:
try:
# 按行执行 SQL 语句, 注意, 这里使用 字符串的format方法将数据替换进去
# 如果可以相信数据的安全性可以这样做, 如果数据来自互联网, 需要用另一种更加安全的方式
cur.execute(artist_insert, {"id": row['id'], "name": row['name'] } )
except Exception as e:
print(e)
print(" 成功导入 CSV 数据" )
# 提交事务
cn.commit()
# 关闭数据库
cn.close()
Step3: 将数据表与计算结果导出为 CSV 文件
接下来我们使用 execute 方法获取数据,然后输出到 CSV 文件,与 SQLite 代码类似:
# 导入 pymysql 模块
import pymysql
# 连接数据库
cn = pymysql.connect(host='localhost', user='pandas', password='pandas', db='pandas')
# 新建游标对象
cur = cn.cursor()
# 用 with语法打开文件
with open("artist_data.csv", 'w') as csvfile:
# 首先读取 description 写入 CSV 的第一行
# 执行 SQL 语句
results= cur.execute("select * from artist")
# 获得 description
description = cur.description
# 新建 CSV writer 对象
writer = csv.writer(csvfile)
# 写入列名称
writer.writerow([ x[0] for x in description ] )
# 遍历数据
for row in cur:
# 写入每一行数据
writer.writerow(row)
print(" 成功写入 CSV 文件")
以上我们使用 PyMySQL 模块为例,介绍了如何使用 Python 对 MySQL 数据库进行增删改查,大家都学会了吗?
— 申请免费试学名额 —
在职想转行提升,担心学不会?根据个人情况规划学习路线,闯关式自适应学习模式保证学习效果
讲师一对一辅导,在线答疑解惑,指导就业!
相关推荐 更多
人工智能python入门基础教程学哪些内容?
人工智能python入门基础教程学哪些内容?Python课程内容:在python中使用数字、编程中的逻辑、在字符串中存储文本、信息的输入和输出、列表、使用循环重复代码、创建简单的脚本等。初学者可以在网站找到很多的免费资源。
6928
2019-04-10 00:08:19
2019年Python学习路线及学习目标规划拿走不谢
2019年Python学习路线及学习目标规划拿走不谢,Python作为今年来特别受欢迎的编程语言,是AI时代头牌语言AI领域的敲门砖,Python已经入驻小学生教材,将来不学Python不仅知识会脱节与小朋友都没共同话题了,作为程序员的我们,必须给自己增加一项技能,提高职场竞争力,掌握一定的Python技能。
4711
2019-09-27 15:18:28
为什么学习Python?
近年来,不论是在职的程序员群体,还是新入行互联网的新同学,越来越多的人开始学习Python,而对于很多没有接触过Python的同学而言,总是会提问:为什么学习Python?针对不同的人群学习Python的需求是不同的。
2146
2020-05-15 17:49:21
入门Python学习难吗?怎样规划学习路线?
入门Python学习难吗?怎样规划学习路线?Python是一种跨平台的计算机程序设计语言。Python现在在各个邻域都有十分广泛的应用,在近几年的上升趋势十分明显,未来的发展前景也十分广阔。Python入门不难,但入门后不断学习是十分重要的,相对来说较为漫长。
2416
2020-06-08 11:17:25
循环语句是什么?起什么作用?
循环语句是什么?起什么作用?循环语句由循环体及循环的终止条件两部分组成。在不少实际问题中有很多有规律性的重复操作,在程序中就需要重复执行某些语句。编程语言中的循环语句包括三种,分别是for,while,do……while
932
2021-12-28 17:01:04
热门文章
- Kafka消息的分发流程是什么?
- 比较实用的大数据分析模型有哪些?
- Java原子操作之CAS原子指令学习
- 从后端开发转大数据开发怎么样?
- 30条实用MySQL优化法则
- 目前Java架构师现状如何?前景怎么样?
- 大数据行业现在工作好不好找?很难吗?
- 和行业大咖1V1免费聊IT,这是头一次!
- 人工智能如何入门学习?前景如何
- 未来互联网人才还稀缺吗?哪些技术方向热门? 查看更多
扫描二维码,了解更多信息
