结合几个热图

Question

我有几个矩阵，我想用不同的颜色显示它们，但在一个热图中。例如：

import numpy as np
a=np.array([[0,0,0],[0,0,0],[1,0,0]])
b=np.array([[0,0,0],[2,2,0],[0,0,0]])
c=np.array([[3,0,0],[0,3,0],[0,0,3]])

a+b+c=np.array([[3, 0, 0],
   [2, 5, 0],
   [1, 0, 3]])

。我想在一个热图中根据 1、2、3 为非零位置赋予不同的颜色。 a 和 b 没有问题，但 b 和 c 有重叠。那么我怎样才能清楚地显示它们呢？ 所以对于最终的矩阵a+b+c，我希望人们能理解1发生在3,1位置，一种颜色，2发生在2,1和2,2，另一种颜色，3发生在位置1,1和2 ,2 和 3,3 与第三种颜色。

Answer 1

我仍然对你想要实现的目标有点困惑，但我希望这就是你想要的

我已将每个数组分配给输出图像中的一个通道（红色、绿色和蓝色） 这给出了以下颜色图：

• 没有值：黑色
• a、b 和 c 的值：白色
• 只有一个：红色
• 只有 b：绿色
• a 和 b 的值是：黄色
• 等等……

这里是sn-p：

import numpy as np
from matplotlib import pyplot as plt

# input data
a = np.array([[0,0,0],[0,0,0],[1,0,0]])
b = np.array([[0,0,0],[2,2,0],[0,0,0]])
c = np.array([[3,0,0],[0,3,0],[0,0,3]])

# transform the data so that all values are either 0 or one
a = a.astype(bool)
b = b.astype(bool)
c = c.astype(bool)

# create the empty output image (heatmap)
width, height = a.shape
img = np.zeros((width, height, 3))
# put the data into the image
img[:,:,0] = a
img[:,:,1] = b
img[:,:,2] = c

plt.imshow(img)
plt.show()

结果如下：

希望对你有帮助