2.8 从数据库导入数据
1、准备工作
(1)首先需要安装SQLite数据库pip install sqlite3
(2)可以在IPython中执行下述代码来验证一下是否都已经安装好
(3)
import sqlite3
sqlite3.version
sqlite3.sqlite_version
2、操作步骤
从数据库读取数据,需要以下步骤
(1)连接数据库引擎
(2)在选择的表上执行查询操作
(3)读取从数据库引擎返回的结果
对于SQL语句的使用,下面举一个简单的例子:
SELECT ID, Name, Population FROM City ORDER BY Population
DESC LIMIT 1000
这条语句从City表中查询了ID、Name和Population等列(字段)的值。ORDER BY告诉数据库引擎按照Population列对数据进行排序,同时DESC指定按降序排列
3、我们将使用 world.sql 示例中的表。这个表包含了全
世界的城市名和人口,有超过5000条的数据。
4、首先需要把这个SQL文件导入到SQLite数据库中
import sqlite3
import sys
db = "C:/Users/18902/PycharmProjects/python-data-visualization/02_Code/test.db"
script_path="C:/Users/18902/PycharmProjects/python-data-visualization/02_Code/world.sql"
try:
#打开数据库test
con = sqlite3.connect(db)
with con:
#创建一个游标
cur = con.cursor()
#将SQL文件导入SQLite数据库中
with open(script_path,'r') as f:
cur.executescript(f.read())
except sqlite3.Error as err:
print("Error occured: %s" % err)
上面这段代码会读取 SQL 文件中的 SQL 语句,然后在打开的 SQLite db文件上执行。
在把数据导入到数据库之后,就能查询数据并进行一些操作了。以下是从数据库文件读取数据的代码。
import sqlite3
import sys
db = "C:/Users/18902/PycharmProjects/python-data-visualization/02_Code/test.db"
try:
con = sqlite3.connect(db)
with con:
#使用游标对象SQL语句查询
cur = con.cursor()
query = 'SELECT ID, Name, Population FROM City ORDER BY Population DESC LIMIT 1000'
con.text_factory = str
#执行SQL语句
cur.execute(query)
#从结果中取出所有记录
#fetchone()--从结果中取一条记录,并将游标指向下一条记录
#fetchmany()--从结果中取多条记录
resultset = cur.fetchall()
# extract column names
col_names = [cn[0] for cn in cur.description]
print("%10s %30s %10s" % tuple(col_names))
print("="*(10+1+30+1+10))
for row in resultset:
print("%10s %30s %10s" % row)
except sqlite3.Error as err:
print("[ERROR]:", err)
5、运行测试
6、解析
(1)连接成功,我们通过con.cursor()得到一个游标。游标与迭代器类似,能让我们遍历数据库返回的结果集中的记录。
(2)我们定义了一个查询操作,与数据库建立连接后,执行查询请求并通过cur.fetchall()得到结果集。如果只想获取一条结果,可以用fetchone()。
(3)在cur.description上执行列表解析操作来得到数据库的列名。description是一个只读属性,包含了很多的信息。对每一列的信息都有一个7个元素的元组,这里只用到列名,所以仅获得每个元组的第一个元素。
7、补充
如果小伙伴们需要数据库test.db和world.sql文件可以到我的主页资源那里下载哦!如果觉得不错的话可以给我点个赞哦!!!