plt.pcolormesh的使用

plt.pcolormesh是Python中用于创建2D数组的快速可视化的函数。它的用法如下：

plt.pcolormesh(X, Y, Z, cmap=None, vmin=None, vmax=None)

其中，X和Y表示网格的坐标，Z表示网格上的值。cmap参数是一个颜色映射表，用于指定绘图中使用的颜色。vmin和vmax参数分别用于指定数据的最小值和最大值。

以下是一个使用plt.pcolormesh绘制2D数组的示例代码：

import numpy as np
import matplotlib.pyplot as plt

# 创建一个10x10的网格
X, Y = np.meshgrid(np.arange(10), np.arange(10))
# 创建一个10x10的随机数组
Z = np.random.rand(10, 10)

# 绘制2D数组
plt.pcolormesh(X, Y, Z, cmap='coolwarm')

# 添加颜色条
plt.colorbar()

# 显示图像
plt.show()

这个示例代码创建了一个10x10的网格，并在每个网格上随机生成一个值。然后，使用plt.pcolormesh函数将这个2D数组绘制出来，并使用'coolwarm'颜色映射表进行着色。最后，使用plt.colorbar函数添加一个颜色条，以便查看颜色对应的值。运行这个代码将会得到一个如下所示的图像：


在这个图像中，越亮的颜色表示越高的值，越暗的颜色表示越低的值。可以看到，使用plt.pcolormesh函数可以快速地将一个2D数组可视化出来，并且可以方便地添加颜色映射表和颜色条。