plt.imshow控制颜色示例

好的，以下是一个plt.imshow控制颜色的示例代码：

import numpy as np
import matplotlib.pyplot as plt

# 创建布朗运动随机矩阵
np.random.seed(0)
z = np.random.randn(10, 10)

# 设定颜色范围
vmin, vmax = z.min(), z.max()

# 设置颜色条
fig, ax = plt.subplots(1, 1, figsize=(7, 4))
cax = ax.imshow(z, interpolation='nearest', vmin=vmin, vmax=vmax, cmap='coolwarm')
fig.colorbar(cax)

plt.show()

以上代码会创建一个随机的10 x 10的矩阵，并使用plt.imshow来绘制矩阵的颜色图，同时设置了颜色条来显示颜色的范围。在这个示例中，我们使用了coolwarm色彩映射(cmap)来控制颜色的变化。