如何在 Python 中更改 Matplotlib 颜色条的大小？

---

可视化工具是 Matplotlib 库的重要组成部分。其中一个工具是颜色条。它显示了绘图中数据值和颜色之间的映射关系。为了调整颜色条的大小以使其更清晰可见或更好地适应绘图，Matplotlib 中的 colorbar() 函数提供了几个参数。

在本文中，我们将讨论如何使用不同的方法更改 matplotlib 颜色条的大小。

什么是 Matplotlib 以及如何安装它？

Matplotlib 是一个广泛用于在 Python 中创建静态、动画和交互式可视化的库。它有一个名为“pyplot”的模块，通过提供控制线型、字体属性、格式化轴等功能，使绘图变得容易。

我们可以使用以下命令安装 matplotlib –

pip install matplotlib

什么是 pyplot？

Pyplot 是 Matplotlib 库的一个子库。它通常用于数据分析和可视化任务，因为它可以帮助用户使用不同的参数轻松更改其绘图的外观。它是一种用更少的代码创建复杂图形的简单方法。它提供了各种选项，例如不同的颜色、标题、线型，允许您向绘图添加文本、图像等。

Matplotlib 中的颜色条？

颜色条是标量值映射的可视化表示。颜色条是一个条形，其中包含各种颜色，放置在 matplotlib 图表的一侧。它是图表中显示的颜色图例。默认情况下，matplotlib 颜色条的位置在 matplotlib 图表的右侧。

颜色条中的颜色映射包含五种颜色。“rainbow”颜色映射在颜色条中包含七种颜色。

matplotlib 的 pyplot 模块中的 colorbar() 函数用于为绘图创建颜色条，指示颜色比例。我们可以更改颜色条的方向，它可以是水平的或垂直的。

示例

在这个绘图中，我们使用 Matplotlib 在散点图中创建了垂直颜色条。命名了两个列表“items”和“sales”，然后使用 plt.scatter() 函数为散点图创建了一个绘图。

X 轴标题为“ITEMS”，Y 轴标题为“SALES”，并将整个图形标题为“SHOP MANGEMENT”。为了在绘图中创建颜色条，我们使用了 plt.colorbar() 函数。

import matplotlib.pyplot as plt
items=["chocolate","toffees","cake","colddrinks","lays","biscuit","frooti"]
sales=[89000,23000,98777,12000,65000,11100,96000]
c=['r','y','m','r','y','m','b']
a=plt.scatter(items,sales,color=c)
plt.xlabel("ITEMS")
plt.ylabel("SALES")
plt.suptitle("SHOP MANGEMENT",fontsize=20)
plt.colorbar()
plt.show()

输出

示例

在这个绘图中，我们通过在 Matplotlib 的 plt.colorbar 函数中使用 orientation=”horizontal” 创建了水平颜色条。

import matplotlib.pyplot as plt
items=["chocolate","toffees","cake","colddrinks","lays","biscuit","frooti"]
sales =[89000,23000,98777,12000,65000,11100,96000]
c= ['r','y','m','r','y','m','b']
a=plt.scatter(items, sales, color=c)
plt.xlabel("ITEMS")
plt.ylabel("SALES")
plt.suptitle("SHOP MANGEMENT", fontsize=20)
plt.colorbar(orientation="horizontal")
plt.show()

输出

热力图的颜色条

以类似的方式，您可以将颜色条添加到不同的绘图中。让我们以热力图绘图为例。我们可以使用 NumPy rand() 函数创建一个 2D 随机值数组。

示例

import matplotlib.pyplot as plt
import numpy as np
data = np.random.rand(4, 4)
heatmap = plt.imshow(data, cmap='Greens')
colorbar = plt.colorbar(heatmap)
plt.show()

输出

更改 Matplotlib 中颜色条的大小？

我们可以通过将值传递给 colorbar() 函数的 shrink 参数来更改颜色条的大小。如果需要减小颜色条的大小，shrink 值应小于 1。根据您传递的值，大小会减小。如果您将 0.5 作为值传递给此参数，则颜色条的大小将减半。

示例

这里，我们使用 plot matplotlib 绘制了一个散点图。对于该图，我们创建了两个列表“items”和“sales”，然后使用 plt.scatter() 语句创建了一个散点图。我们创建了一个 shrink 值为 0.5 的颜色条。

import matplotlib.pyplot as plt
items=["chocolate","toffees","cake","colddrinks","lays","biscuit","frooti"]
sales=[89000,23000,98777,12000,65000,11100,96000]
c= ['r','y','m','r','y','m','b']
a= plt.scatter(items, sales, color=c)
plt.xlabel("ITEMS")
plt.ylabel("SALES")
plt.suptitle("SHOP MANGEMENT",fontsize=20)
plt.colorbar(a, shrink=0.5)
plt.show()

输出

当 shrink 大小为 2 时，输出如下 –

您也可以动态更改 Matplotlib 中颜色条的大小，编写一个函数，该函数将以某种方式工作，即它将获取颜色条的当前大小作为输入值，并根据外部因素（例如数据量、绘图大小）返回一个新大小。

结论

在本文中，我们讨论了什么是颜色条，它是如何创建的以及如何使用各种示例在 matplotlib 中更改颜色条的大小。