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

matplotlibpythondata visualization

可视化工具是 Matplotlib 库的重要组成部分。其中一个工具是 colorbar。它显示图中数据值和颜色之间的映射。为了调整颜色条的大小以使其更明显或更好地与图匹配,我们在 Matplotlib 中的 colorbar() 函数中提供了几个参数。

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

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

Matplotlib 是一个广泛使用的库,用于在 Python 中创建静态、动画和交互式可视化。它有一个模块名称"pyplot",通过提供控制线条样式、字体属性、格式化轴等功能,使绘图变得简单。

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

pip install matplotlib

什么是 pyplot?

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

Matplotlib 中的颜色条?

颜色条是标量值映射的可视化。颜色条是放置在 matplotlib 图表两侧的由各种颜色组成的条形图。它是图表中显示的颜色的图例。默认情况下,matplotlib 颜色条的位置位于 matplotlib 图表的右侧。

颜色条中的颜色图由五种颜色组成。颜色条中的"彩虹"颜色图有七种颜色。

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

示例

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

X 轴标题为"ITEMS",y 轴标题为"SALES",整个图表标题为"SHOP MANGEMENT"。为了在图中创建颜色条,我们使用了 plt.colorbar() 函数。

将 matplotlib.pyplot 导入为 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="horizo​​ntal" 创建了水平颜色条。

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() 函数创建一个随机值的二维数组。

示例

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 中颜色栏的大小?

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

示例

在这里,我们使用 plot matplotlib 绘制了散点图。对于该图,我们创建了两个列表"items"和"sales",然后使用 plt.scatter() 语句创建了散点图。我们创建了一个颜色条,其缩放值为 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()

输出

以下是当收缩尺寸为 2 时的输出 –

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

结论

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


相关文章