如何使用颜色条 Matplotlib 在 Python 中绘制 2D 矩阵?

matplotlibpythondata visualization

要在 Python 中使用颜色条绘制 2D 矩阵,我们可以使用 numpy 创建 2D 数组矩阵,并在 imshow() 方法中使用该矩阵。

步骤

  • 使用 numpy 创建 data2D

  • 使用 imshow() 方法将数据显示为图像,即在 2D 常规栅格上。

  • 使用 colorbar() 方法和 imshow() 标量可映射图像为 ScalarMappable 实例 *mappable* 创建颜色条。

  • 要显示图形,请使用show() 方法。

示例

import numpy as np
from matplotlib import pyplot as plt
plt.rcParams["figure.figsize"] = [7.00, 3.50]
plt.rcParams["figure.autolayout"] = True
data2D = np.random.random((50, 50))
im = plt.imshow(data2D, cmap="copper_r")
plt.colorbar(im)
plt.show()

输出


相关文章