Matplotlib在饼图中分配特定颜色给数据

Matplotlib是一个用于绘制2D图表和图形的Python可视化库。Matplotlib中的饼图以圆形的形式显示数据的相对大小，并将数据分配给简单的颜色。在默认情况下，Matplotlib会自动选择与统计数据匹配的颜色，然而，有时我们需要手动选择某些数据的颜色。在这篇文章中，我们将演示如何在Matplotlib的饼图中分配特定颜色给数据。

阅读更多：Matplotlib 教程

数据准备

在开始之前，我们需要准备我们的数据。我们将使用一个小数据集，其中包含几个谷物产品的销售情况。每个产品销售情况的数据都以百分比的形式给出。我们将想要在饼图中为每个产品分配特定的颜色。

product_list = ['Oatmeal', 'Bran Flakes', 'Corn Flakes', 'Honey Nut Cheerios']
sales_data = [15, 30, 35, 20]

绘制饼图

接下来，我们将使用Matplotlib的pie()函数来绘制饼图。在pie()函数中，我们需要将数据与标签作为参数传递。此外，如果需要手动为数据中的每个部分分配颜色，需要设置colors参数。在这个例子中，我们将使用四种颜色，每一种颜色对应一个产品。

import matplotlib.pyplot as plt

product_list = ['Oatmeal', 'Bran Flakes', 'Corn Flakes', 'Honey Nut Cheerios']
sales_data = [15, 30, 35, 20]
colors = ['lightblue', 'orange', 'lightgreen', 'yellow']

plt.pie(sales_data, labels=product_list, colors=colors)
plt.show()

Matplotlib在饼图中分配特定颜色给数据

在上面的代码中，我们使用lightblue、orange、lightgreen和yellow作为我们产品列表中每个产品的颜色。最终的饼图如上所示。

配置颜色列表

我们可以使用Python的字典来配置每个产品的颜色。在这种情况下，字典中的键值对表示产品与颜色的对应关系。我们可以使用for循环来创建颜色列表，使其与数据列表中的相应元素一一对应。

import matplotlib.pyplot as plt

sales_data = [15, 30, 35, 20]
product_list = ['Oatmeal', 'Bran Flakes', 'Corn Flakes', 'Honey Nut Cheerios']

color_dict = {'Oatmeal':'lightblue', 'Bran Flakes':'orange', 'Corn Flakes':'lightgreen', 'Honey Nut Cheerios':'yellow'}
colors = [color_dict[key] for key in product_list]

plt.pie(sales_data, labels=product_list, colors=colors)
plt.show()

在上面的代码中，我们使用color_dict字典来指定每个产品的颜色。然后，我们使用颜色列表colors来与数据列表sales_data和product_list一起传递给pie()函数。最终的饼图如下所示。

Matplotlib在饼图中分配特定颜色给数据

自定义颜色列表

当我们需要为一组数据手动指定颜色时，我们可以选择使用一组自定义颜色。在这种情况下，我们只需在colors参数中将颜色列表传递给pie()函数。

import matplotlib.pyplot as plt

sales_data = [15, 30, 35, 20]
product_list = ['Oatmeal', 'Bran Flakes', 'Corn Flakes', 'Honey Nut Cheerios']

colors = ['#33FF8A', '#C13EFF', '#F08080', '#00BFFF']

plt.pie(sales_data, labels=product_list, colors=colors)
plt.show()

饼图使用了自定义颜色的样式，如上代码中我们使用了4个十六进制的颜色代码分别为’#33FF8A’, ‘#C13EFF’, ‘#F08080’, ‘#00BFFF’。最终的饼图如下。

Matplotlib在饼图中分配特定颜色给数据

提高代码复用性

如果我们需要在多个图表中使用相同的颜色列表，我们可以选择将颜色列表分离出来并定义为全局列表。在这种情况下，我们可以非常轻松地对颜色进行任何必要的更改。

import matplotlib.pyplot as plt

sales_data = [15, 30, 35, 20]
product_list = ['Oatmeal', 'Bran Flakes', 'Corn Flakes', 'Honey Nut Cheerios']

global_colors = ['#33FF8A', '#C13EFF', '#F08080', '#00BFFF']

plt.pie(sales_data, labels=product_list, colors=global_colors)
plt.show()

在这个例子中，我们定义了一个全局颜色列表global_colors，然后我们只是将它与其他参数一起传递给pie()函数。这可以使代码更加可读性和可维护性。