带有颜色映射的数组到图像

pip install matplotlib

import numpy as np
import matplotlib.pyplot as plt

# 创建一个带有颜色映射的数组
data = np.random.rand(10, 10)  # 生成一个10x10的随机数组

# 使用matplotlib的imshow函数将数组转换为图像
plt.imshow(data, cmap='viridis')  # 你可以使用其他颜色映射，如'plasma', 'inferno', 'magma'等

# 添加颜色条
plt.colorbar()

# 显示图像
plt.show()