2.8 从数据库导入数据

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文件可以到我的主页资源那里下载哦！如果觉得不错的话可以给我点个赞哦！！！