Python连接数据库使用matplotlib画柱形图

下面是Python连接数据库使用matplotlib画柱形图的完整攻略，希望对你有所帮助。

1. 安装数据库模块

在Python中连接数据库，我们需要使用相应的数据库驱动模块。常用的数据库驱动模块包括MySQLdb、pymysql、sqlite3等等，根据不同的数据库选择不同的驱动模块。

以连接MySQL数据库为例，我们可以使用pymysql模块，通过以下命令安装：

pip install pymysql

2. 连接数据库

在安装好数据库模块之后，我们需要通过代码来连接数据库。具体的连接方式和参数根据不同的数据库而有所不同，可以参考相应的驱动模块文档。

以连接MySQL数据库为例，我们可以使用以下代码：

import pymysql

# 打开数据库连接
db = pymysql.connect(host='localhost', port=3306, user='root', passwd='password', db='testdb', charset='utf8mb4')

# 使用 cursor() 方法创建一个游标对象 cursor
cursor = db.cursor()

# 执行 SQL 查询
cursor.execute("SELECT * FROM student")

# 获取所有记录列表
results = cursor.fetchall()

# 关闭数据库连接
db.close()

以上代码中，我们先通过pymysql.connect()函数来连接指定的MySQL数据库，其中参数包括host、port、user、passwd、db、charset等。

然后，我们使用cursor()函数来创建一个游标对象，通过游标对象我们可以执行SQL查询语句，获得查询的结果。

最后，我们通过db.close()函数来关闭数据库连接。

3. 使用matplotlib绘制柱状图

在连接成功数据库并获取了查询结果之后，我们可以使用matplotlib模块来绘制柱状图。具体的绘图过程可以参考matplotlib的官网文档。

以下是一个简单的柱状图绘制示例：

import matplotlib.pyplot as plt

# 模拟数据
height = [3, 12, 5, 18, 45, 3, 25]
bars = ('A', 'B', 'C', 'D', 'E', 'F', 'G')
y_pos = range(len(bars))

# 创建 Figure 对象
fig = plt.figure(figsize=(8, 6))

# 创建 Axes 对象
ax = fig.add_subplot(111)

# 绘制柱形图
ax.bar(y_pos, height, align='center', alpha=0.5)

# 设置x轴、y轴的标签、范围、标题等属性
plt.xticks(y_pos, bars)
plt.ylabel('Number')
plt.xlabel('Letters')
plt.title('Bar chart')

# 显示图形
plt.show()

以上代码中，我们定义了一些模拟数据，包括每个柱形的高度height、柱形的标签bars、y轴标签Number、x轴标签Letters和标题Bar chart。

然后，我们使用fig.add_subplot()函数来创建Axes对象，通过它来绘制柱状图。在绘制柱状图时，我们使用了ax.bar()函数，其中包括柱形的高度、标签和对齐方式等属性。

最后，我们设置了x轴、y轴的标签、范围、标题等属性，并使用plt.show()函数来显示图形。

4. 将数据库结果连同柱状图结合起来

当我们连接成功数据库并查询到了相应的结果，就可以将其与柱状图结合起来。具体的方法包括以下几个步骤：

1. 将结果转换为列表或字典形式；
2. 根据结果绘制柱状图，将柱形标签设置为结果中的某些字段，柱形高度设置为结果中的另外一些字段；
3. 显示图形。

以下是一个完整的示例：

import pymysql
import matplotlib.pyplot as plt

# 连接数据库
db = pymysql.connect(host='localhost', port=3306, user='root', passwd='password', db='testdb', charset='utf8mb4')

# 获取游标对象
cursor = db.cursor()

# 执行查询
cursor.execute("SELECT city, population FROM city_population")

# 获取查询结果
results = cursor.fetchall()

# 将结果转换为列表形式
data = [(r[0], r[1]) for r in results]

# 关闭数据库连接
db.close()

# 绘制柱状图
bars = [d[0] for d in data]
heights = [d[1] for d in data]

fig = plt.figure(figsize=(8, 6))
ax = fig.add_subplot(111)
ax.bar(range(len(bars)), heights, align='center', alpha=0.5)
plt.xticks(range(len(bars)), bars)
plt.ylabel('Population')
plt.xlabel('City')
plt.title('Population of Cities')

# 显示图形
plt.show()

以上代码中，我们首先连接了本地的MySQL数据库，并查询了city_population表的city和population字段。然后，我们将查询结果转换为一个列表形式，每个元素为一个元组，其中第一个值为城市名称，第二个值为城市的人口数。

最后，我们使用ax.bar()函数绘制了柱状图，其中柱形标签设置为城市名称，柱形高度设置为城市的人口数。我们也可以在图中添加各种标签、范围、标题等属性，来使图像更加直观和易于理解。