饼图 matplotlib 图例 - Python

在进行数据分析和可视化时，饼图是一种常见的图表类型。本文将介绍如何使用 Python 中的 matplotlib 库绘制饼图，并添加图例。

准备数据

首先，让我们准备一些数据来绘制饼图。假设我们正在分析一个小企业的收入来源，收入来自三个渠道：销售、服务和广告。我们可以用以下代码创建一个包含这三个渠道和相应收入的字典：

data = {'Sales': 245000, 'Services': 120000, 'Advertising': 55000}

绘制饼图

接下来，我们需要使用 matplotlib 库的 pyplot 模块来创建饼图。可以使用以下代码：

import matplotlib.pyplot as plt

# 创建一个子图并指定大小
fig, ax = plt.subplots(figsize=(6, 6))

# 绘制饼图
ax.pie(data.values(), labels=data.keys())

# 显示图表
plt.show()

这将绘制一个包含三个数据点的饼图，每个数据点代表一个渠道。饼图上每个扇形的面积表示该渠道在总收入中所占的比例。

添加图例

为了使饼图更加易于理解，我们可以添加一个图例，说明每个扇形代表哪个渠道。可以使用 legend 函数来添加图例。以下是添加图例的完整代码：

import matplotlib.pyplot as plt

# 创建一个子图并指定大小
fig, ax = plt.subplots(figsize=(6, 6))

# 绘制饼图
wedges, labels = ax.pie(data.values(), labels=data.keys())

# 添加图例
ax.legend(wedges, data.keys(), title='Channels', loc='center left', bbox_to_anchor=(1, 0, 0.5, 1))

# 显示图表
plt.show()

在上面的代码中，我们将 pie 函数的返回值（扇形和标签）传递给 legend 函数。我们还指定了图例的标题、位置和边界框。

结论

在本文中，我们介绍了如何使用 Python 中的 matplotlib 库绘制饼图，并添加了图例。饼图和图例是可视化分析的重要元素，可以使数据更加易于理解和解释。