Python 绘制柱状图|极客教程

Python 绘制柱状图

在数据可视化中，柱状图是一种常用的图表类型，用于展示不同类别之间的数量或比较。Python中的matplotlib库提供了丰富的功能，可以轻松绘制各种图表，包括柱状图。

1. 导入必要的库

在绘制柱状图之前，首先需要导入matplotlib库。如果你还没有安装matplotlib库，可以使用以下命令进行安装：

pip install matplotlib

导入库的代码如下：

import matplotlib.pyplot as plt

2. 基本的柱状图

下面我们来看一个简单的示例，展示不同水果的销售量。

# 数据
fruits = ['苹果', '香蕉', '橙子', '西瓜']
sales = [100, 80, 120, 150]

# 绘制柱状图
plt.bar(fruits, sales)

# 添加标题和标签
plt.title('不同水果销售量')
plt.xlabel('水果')
plt.ylabel('销售量')

# 显示图形
plt.show()

运行以上代码，我们可以得到一个简单的柱状图，如下所示：

柱状图示例图像

3. 自定义柱状图样式

除了基本的柱状图外，我们还可以对柱状图进行自定义，包括柱子的颜色、宽度、间距等。

# 数据
fruits = ['苹果', '香蕉', '橙子', '西瓜']
sales = [100, 80, 120, 150]

# 自定义样式
colors = ['r', 'g', 'b', 'y']
width = 0.5

# 绘制柱状图
plt.bar(fruits, sales, color=colors, width=width)

# 添加标题和标签
plt.title('不同水果销售量')
plt.xlabel('水果')
plt.ylabel('销售量')

# 显示图形
plt.show()

在上面的代码中，我们指定了柱子的颜色为红色、绿色、蓝色和黄色，宽度为0.5。运行以上代码，我们可以看到柱状图的样式发生了变化。

4. 绘制多组数据

有时候我们需要在同一个柱状图上展示多组数据，可以使用plt.bar和plt.xticks方法来实现。

# 数据
fruits = ['苹果', '香蕉', '橙子', '西瓜']
sales_2019 = [100, 80, 120, 150]
sales_2020 = [120, 90, 110, 160]

# 设置每组柱子的宽度
bar_width = 0.35

# 绘制柱状图
plt.bar([i - bar_width/2 for i in range(len(fruits))], sales_2019, width=bar_width, label='2019')
plt.bar([i + bar_width/2 for i in range(len(fruits))], sales_2020, width=bar_width, label='2020')

# 添加标题和标签
plt.title('不同水果销售量对比')
plt.xlabel('水果')
plt.ylabel('销售量')
plt.xticks(range(len(fruits)), fruits)

# 显示图例
plt.legend()

# 显示图形
plt.show()

运行以上代码，我们可以看到不同年份的销售量在同一个柱状图上进行了对比。

5. 水平柱状图

除了垂直柱状图外，我们还可以绘制水平柱状图，只需要使用plt.barh方法即可。

# 数据
fruits = ['苹果', '香蕉', '橙子', '西瓜']
sales = [100, 80, 120, 150]

# 绘制水平柱状图
plt.barh(fruits, sales)

# 添加标题和标签
plt.title('不同水果销售量')
plt.xlabel('销售量')
plt.ylabel('水果')

# 显示图形
plt.show()

以上代码中，我们使用plt.barh方法绘制了水平柱状图，实现了水果销售量的可视化。