如何在Matplotlib中为subplot添加colorbar|极客教程

如何在Matplotlib中为subplot添加colorbar

参考：how to add colorbar to subplot matplotlib

Matplotlib是一个广泛使用的Python绘图库，可以创建各种类型的图表，包括散点图、折线图和热图等。在Matplotlib中，我们可以使用subplot来创建多个子图，并且可以为每个子图添加颜色条。

1. 创建多个子图

要创建多个子图，我们可以使用subplot函数。以下是一个简单的示例代码：

import matplotlib.pyplot as plt

# 创建第一个子图
plt.subplot(2, 2, 1)
plt.plot([1, 2, 3, 4], [1, 4, 9, 16])

# 创建第二个子图
plt.subplot(2, 2, 2)
plt.plot([1, 2, 3, 4], [1, 2, 3, 4])

plt.show()

Output:

如何在Matplotlib中为subplot添加colorbar

在上面的示例中，我们使用subplot函数创建了一个2×2的网格，并向其添加了两个子图。现在我们将向这些子图中添加颜色条。

2. 添加颜色条

要向子图添加颜色条，我们可以使用colorbar函数。以下是一个简单的示例代码：

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 10, 100)
y = np.sin(x)

plt.subplot(2, 1, 1)
plt.scatter(x, y, c=y, cmap='viridis')
plt.colorbar()

plt.subplot(2, 1, 2)
plt.plot(x, y)
plt.colorbar()

plt.show()

Output:

如何在Matplotlib中为subplot添加colorbar

在上面的示例中，我们在第一个子图中绘制了散点图，并添加了颜色条。在第二个子图中绘制了折线图，并添加了颜色条。

3. 在不同的子图中添加不同的颜色条

有时我们可能需要在不同的子图中使用不同的颜色条。以下是一个示例代码：

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 10, 100)
y1 = np.sin(x)
y2 = np.cos(x)

plt.subplot(2, 1, 1)
sc = plt.scatter(x, y1, c=y1, cmap='viridis')
plt.colorbar(sc)

plt.subplot(2, 1, 2)
lc = plt.plot(x, y2, c=y2, cmap='plasma')[0]
plt.colorbar(lc)

plt.show()

在上面的示例中，我们在两个子图中分别绘制了不同的图表，并使用不同的颜色条。

4. 使用不同的颜色映射

除了使用默认的颜色映射之外，我们还可以自定义颜色映射。以下是一个示例代码：

import numpy as np
import matplotlib.pyplot as plt
from matplotlib.colors import ListedColormap

x = np.random.rand(100)
y = np.random.rand(100)
colors = np.random.rand(100)

cmap = ListedColormap(['red', 'green', 'blue'])

plt.scatter(x, y, c=colors, cmap=cmap)
plt.colorbar()

plt.show()

Output:

如何在Matplotlib中为subplot添加colorbar

在上面的示例中，我们自定义了一个包含红色、绿色和蓝色的颜色映射，并将其应用于散点图。

5. 隐藏颜色条

有时，我们可能希望隐藏颜色条。以下是一个示例代码：

import numpy as np
import matplotlib.pyplot as plt

x = np.linspace(0, 10, 100)
y = np.sin(x)

plt.scatter(x, y, c=y, cmap='viridis')
cb = plt.colorbar()
cb.remove()

plt.show()