python代码热力图中数值为0的颜色和批注数字不显示
以下是一个使用Python绘制热力图的示例代码,可以将数值为0的颜色和批注数字隐藏:
import numpy as np
import matplotlib.pyplot as plt
# 生成随机数据
data = np.random.rand(10, 10)
# 将数值为0的颜色设置为白色,隐藏批注数字
cmap = plt.cm.get_cmap('jet')
cmap.set_bad(color='white')
# 绘制热力图
plt.imshow(data, cmap=cmap, vmin=0.0001, vmax=1)
plt.colorbar()
# 隐藏x、y轴标签
plt.xticks([])
plt.yticks([])
plt.show()
在上面的代码中,我们使用numpy生成了一个10x10的随机矩阵,然后使用matplotlib的imshow函数绘制了热力图。为了将数值为0的颜色和批注数字隐藏,我们使用了get_cmap函数获取了颜色映射对象,然后使用set_bad方法将数值为0的颜色设置为白色。最后,我们使用xticks和yticks函数隐藏了x、y轴标签。
原文地址: http://www.cveoy.top/t/topic/bd9t 著作权归作者所有。请勿转载和采集!