使用 PyMySQL 查询数据库并把结果保存为列表的步骤如下:
- 安装 PyMySQL 库
使用 pip
命令安装 PyMySQL 库:
pip install PyMySQL
- 连接数据库
使用 pymysql.connect()
方法连接 MySQL 数据库:
import pymysql
# 打开数据库连接
db = pymysql.connect(host="localhost", user="root", password="password", database="testdb")
# 使用 cursor() 方法创建一个游标对象 cursor
cursor = db.cursor()
# SQL 查询语句
sql = "SELECT * FROM customers"
# 执行 SQL 语句
cursor.execute(sql)
# 获取所有记录列表
results = cursor.fetchall()
其中,host
为数据库所在服务器的 IP 或主机名,user
为数据库账户用户名,password
为数据库账户密码,database
为要连接的数据库名。
- 把结果保存为列表
使用 fetchall()
方法获取所有查询结果,并把结果保存为一个列表,每行数据保存为列表中的一个元素:
# 获取所有记录列表
results = cursor.fetchall()
# 把结果保存为列表
data = []
for row in results:
data.append(list(row))
- 获取指定元素下标
可以使用列表索引来获取指定元素下标。例如,要获取第一行第二列的数据,可以使用 data[0][1]
来访问:
# 获取第一行第二列的数据
print(data[0][1])
下面是一个完整的示例代码:
import pymysql
# 打开数据库连接
db = pymysql.connect(host="localhost", user="root", password="password", database="testdb")
# 使用 cursor() 方法创建一个游标对象 cursor
cursor = db.cursor()
# SQL 查询语句
sql = "SELECT * FROM customers"
# 执行 SQL 语句
cursor.execute(sql)
# 获取所有记录列表
results = cursor.fetchall()
# 把结果保存为列表
data = []
for row in results:
data.append(list(row))
# 获取第一行第二列的数据
print(data[0][1])
# 关闭数据库连接
db.close()
另外,如果要查询指定条件的数据,可以使用 WHERE
关键字来筛选数据,例如:
# 查询年龄大于 30 岁的客户
sql = "SELECT * FROM customers WHERE age > 30"
# 执行 SQL 语句
cursor.execute(sql)
# 获取所有记录列表
results = cursor.fetchall()
# 把结果保存为列表
data = []
for row in results:
data.append(list(row))
以上就是使用 PyMySQL 查询数据库、把结果保存为列表并获取指定元素下标的完整攻略。
本站文章如无特殊说明,均为本站原创,如若转载,请注明出处:使用pymysql查询数据库,把结果保存为列表并获取指定元素下标实例 - Python技术站