python画簇状柱形图与折线图组合

原创

mob64ca12e60047 2024-04-03 06:42:02 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12e60047的原创作品，请联系作者获取转载授权，否则将追究法律责任

Python画簇状柱形图与折线图组合

在数据可视化中，簇状柱形图和折线图是常用的图表类型。簇状柱形图可以用来展示不同类别之间的数据对比，而折线图则可以展示数据的趋势变化。本文将介绍如何使用Python中的Matplotlib库来画簇状柱形图与折线图的组合。

准备工作

在开始之前，首先需要安装Matplotlib库。可以使用以下命令来安装：

pip install matplotlib

接下来，准备一些测试数据。假设我们有以下数据，分别表示不同月份的销售额和利润率：

months = ['January', 'February', 'March', 'April', 'May']
sales = [10000, 12000, 11000, 13000, 14000]
profit_margin = [0.1, 0.12, 0.11, 0.13, 0.14]

画簇状柱形图与折线图组合

首先，我们需要导入Matplotlib库，并设置图形的风格：

import matplotlib.pyplot as plt

plt.style.use('ggplot')

然后，我们可以开始画簇状柱形图。簇状柱形图的关键是要设置柱形的宽度，并调整柱形的位置。我们可以使用numpy库来帮助生成柱形的位置：

import numpy as np

bar_width = 0.35
r1 = np.arange(len(months))
r2 = [x + bar_width for x in r1]

plt.bar(r1, sales, color='b', width=bar_width, edgecolor='grey', label='Sales')
plt.bar(r2, profit_margin, color='r', width=bar_width, edgecolor='grey', label='Profit Margin')

plt.xlabel('Months')
plt.ylabel('Amount')
plt.xticks([r + bar_width/2 for r in range(len(months))], months)
plt.legend()

plt.show()

接下来，我们可以添加折线图。折线图的关键是要使用plot函数，并设置线条的样式：

plt.plot(r1, sales, color='b', marker='o', label='Sales')
plt.plot(r1, profit_margin, color='r', marker='o', label='Profit Margin')

plt.xlabel('Months')
plt.ylabel('Amount')
plt.xticks([r for r in range(len(months))], months)
plt.legend()

plt.show()

最后，我们可以将簇状柱形图和折线图组合在一起。可以通过在同一个坐标轴上绘制柱形图和折线图来实现：

fig, ax1 = plt.subplots()

ax2 = ax1.twinx()
ax1.bar(r1, sales, color='b', width=bar_width, edgecolor='grey', label='Sales')
ax2.plot(r1, profit_margin, color='r', marker='o', label='Profit Margin')

ax1.set_xlabel('Months')
ax1.set_ylabel('Sales', color='b')
ax2.set_ylabel('Profit Margin', color='r')

plt.xticks([r for r in range(len(months))], months)

plt.show()

至此，我们成功地画出了簇状柱形图和折线图的组合。通过这种方式，我们可以直观地展示数据之间的对比和趋势变化。