在 Matplotlib 下显示两个颜色条

在 Matplotlib 中，我们可以使用 colorbar() 函数来创建和添加颜色条。默认情况下，将为图表中的每种颜色映射创建一个颜色条。但是，有时我们可能需要在同一个图表中添加两个不同的颜色条以突出不同的数据范围或数据类型。

以下是如何使用 Matplotlib 显示两个颜色条的代码片段：

import matplotlib.pyplot as plt
import numpy as np

# Create a figure and two subplots
fig, axs = plt.subplots(nrows=1, ncols=2, figsize=(8, 4))

# Generate sample data
x = np.arange(0, 10, 0.1)
y1 = np.sin(x)
y2 = np.cos(x)

# Plot the first function and add a colorbar
im1 = axs[0].plot(x, y1, c=y1, cmap='viridis')
fig.colorbar(im1, ax=axs[0])

# Plot the second function and add a second colorbar
im2 = axs[1].plot(x, y2, c=y2, cmap='plasma')
fig.colorbar(im2, ax=axs[1])

# Show the plot
plt.show()

解释：

我们首先导入了 Matplotlib 库和一个用于生成示例数据的 NumPy 库。然后，我们使用 subplots() 函数创建一个有两个子图的图形。对于每个子图，我们使用 plot() 函数绘制一个函数，并使用不同的颜色映射指定了每个点的颜色。我们使用颜色条函数在每个子图中添加一个颜色条，将其与绘制的函数相关联。最后，我们使用 show() 函数显示图表。

在此示例中，我们使用了两个不同的颜色映射—— viridis 和 plasma，以突出不同的数据范围。您可以使用其他颜色映射或将它们替换为自己的映射名称。