Python 操纵 SQL 提取数据

在现代数据分析和处理中,SQL 是一个非常常用的工具,用于在数据库中查询和操作数据。而 Python 作为一门功能强大的编程语言,也提供了许多库和工具来操作 SQL 数据库。本文将介绍如何使用 Python 操纵 SQL 数据库提取数据,并通过代码示例演示。

1. 安装必要的库

在开始之前,首先需要安装几个必要的库来操作 SQL 数据库。Python 的 sqlite3 库是一个轻量级的数据库引擎,可以用来操作 SQLite 数据库。安装方法如下:

pip install sqlite3

2. 连接数据库并提取数据

接下来,我们将通过 Python 来连接数据库并提取数据。首先,我们需要创建一个 SQLite 数据库并插入一些数据,以便后续操作。

import sqlite3

# 连接数据库
conn = sqlite3.connect('example.db')
c = conn.cursor()

# 创建表格
c.execute('''CREATE TABLE IF NOT EXISTS stocks
             (date text, trans text, symbol text, qty real, price real)''')

# 插入数据
c.execute("INSERT INTO stocks VALUES ('2022-09-06', 'BUY', 'AAPL', 100, 202.55)")
c.execute("INSERT INTO stocks VALUES ('2022-09-07', 'SELL', 'GOOGL', 50, 3010.25)")

# 提交更改并关闭连接
conn.commit()
conn.close()

3. 查询数据并展示

现在我们已经在数据库中插入了一些数据,接下来我们可以使用 Python 查询数据并展示。下面的代码示例演示了如何查询数据库中的数据,并使用 matplotlib 库绘制一个饼状图展示数据。

import sqlite3
import matplotlib.pyplot as plt

# 连接数据库
conn = sqlite3.connect('example.db')
c = conn.cursor()

# 查询数据
c.execute("SELECT * FROM stocks")
rows = c.fetchall()

# 关闭连接
conn.close()

# 处理数据
symbols = [row[2] for row in rows]
qtys = [row[3] for row in rows]

# 绘制饼状图
plt.figure(figsize=(6,6))
plt.pie(qtys, labels=symbols, autopct='%1.1f%%')
plt.title('Stocks Distribution')
plt.show()

总结

通过以上代码示例,我们学习了如何使用 Python 操纵 SQL 数据库提取数据。首先,我们连接数据库并插入数据,然后查询数据并展示。在实际应用中,可以根据具体需求进行更复杂的查询和数据处理操作。Python 提供了丰富的库和工具,使得操作 SQL 数据库变得更加方便和高效。希望本文对您有所帮助,谢谢阅读!