python热力图更换颜色

要更改Python热力图的颜色，您可以使用matplotlib库中的colormap。以下是一个示例代码，它使用viridis colormap替换默认的jet colormap：

import numpy as np
import matplotlib.pyplot as plt

# 创建数据
data = np.random.rand(10, 10)

# 绘制热力图（使用viridis colormap）
heatmap = plt.pcolor(data, cmap='viridis')

# 添加颜色栏
plt.colorbar(heatmap)

# 显示图形
plt.show()

在这里，我们使用了`cmap`参数来指定使用的colormap。您可以从matplotlib文档中查找可用的colormap，或者使用以下命令查看所有可用的colormap：

import matplotlib.cm as cm

print(cm.cmap_d.keys())

这将输出一个包含所有可用colormap名称的列表。您可以将其替换为`cmap`参数的值来更改热力图的颜色。