pyspark csv sql操作 pymysql csv

操作mysql：需要导入pymysql模块

参考代码：

import pymysql
# 打开数据库连接
db = pymysql.connect('123.123.0.126','root','root','fdgfd')
# 使用cursor()方法创建一个游标对象 cursor
cursor = db.cursor()
# execute(）方法执行sql查询
cursor.execute('SELECT VERSION()')
# 使用fetchone()方法获取单条数据
data = cursor.fetchone()
print("Database version : %s" %data)
# 关闭数据库连接
db.close()

结果：

 

 

 

 

 将查询出来的内容写入csv文件中，需导入csv、codecs

codecs是专门用作编码转换

import pymysql
import csv
import codecs
# 打开数据库连接
db = pymysql.connect('123.123.0.126','root','root','fgdjd')
# 使用cursor()方法创建一个游标对象 cursor
cursor = db.cursor()
sql = "SELECT * FROM security_module"
cursor.execute(sql)
data = cursor.fetchone()
print(data)
url ="C:\\Users\\gh8\\Desktop\\user1.csv"
csv_file = csv.writer(codecs.open(url ,'w','utf_8_sig'),dialect ='excel')
csv_file.writerow(['用户名','密码','验证码','结果'])
csv_file.writerow(data)
db.close()