假设我正在使用imshow
在简单的matplotlib热图上绘制以下numpy数据数组;在某些情况下,值将为0.0
。有什么方法可以为显示该值的单元格设置特定颜色?例如,当值为0时,该单元格的颜色必须为黑色
a = np.array([[0.8, 2.4, 2.5, 3.9, 0.0, 4.0, 0.0],
[2.4, 0.0, 4.0, 1.0, 2.7, 0.0, 0.0],
[1.1, 2.4, 0.8, 4.3, 1.9, 4.4, 0.0],
[0.6, 0.0, 0.3, 0.0, 0, 0.0, 0.0],
[0.7, 1.7, 0.6, 2.6, 2.2, 6.2, 0.0],
[0, 1.2, 0.0, 0.0, 0.0, 3.2, 5.1],
[0.1, 2.0, 0.0, 1.4, 0.0, 1.9, 6.3]])
Map = ax.imshow(a, interpolation='none', cmap='coolwarm')
答案 0 :(得分:2)
也许不是完美的解决方案,但绝对是一个简单的解决方案。如果仅是为了创建图像,则可以修改原始数据(或其副本)并将0.0
替换为NaN
。然后,您可以使用set_bad
获得所需的输出。
import matplotlib.pyplot as plt
import matplotlib.cm as cm
import numpy as np
a = np.array([[0.8, 2.4, -2.5, 3.9, 0.0, 4.0, 0.0],
[2.4, 0.0, 4.0, 1.0, 2.7, 0.0, 0.0],
[-1.1, 2.4, 0.8, 4.3, 1.9, 4.4, 0.0],
[0.6, 0.0, 0.3, 0.0, 0, 0.0, 0.0],
[0.7, 1.7, -0.6, 2.6, 2.2, 6.2, 0.0],
[0, 1.2, 0.0, 0.0, 0.0, 3.2, 5.1],
[0.1, 2.0, 0.0, 1.4, 0.0, 1.9, 6.3]])
c_map = cm.get_cmap('rainbow')
c_map.set_bad('k')
b = a.copy()
b[b==0] = np.nan
fig = plt.imshow(b, interpolation='none', cmap=c_map)
plt.colorbar(fig)