python - 在 2d 直方图中设置 bin 颜色(多色图)

Question

我正在尝试创建一个覆盖三个独立数据集的二维直方图。我的想法是将数据集着色为红色、绿色和蓝色，以便容器中红色数据的密度对应于该容器颜色的红色值，蓝色和绿色也是如此。

有examples但据我所知，matplotlib 中没有实现多色绘图。

获取表单的网格

[[ (r,g,b) , (r,g,b) ... (r,g,b) ] ,
 [ (r,g,b) , (r,g,b) ...         ] , 
   .......                   
 [ (r,g,b) , (r,g,b) ... (r,g,b) ]]

完全没有问题。问题是我发现的所有绘图函数都希望将单个 bin 值映射到色标上，它们不允许我设置 bin 的完整颜色值。

我应该寻找一些原始的东西吗？是否已经有直方图可以做到这一点？

Answer 1

如果您的数据已经采用 RGB 格式(权重介于 0 和 1 之间)，则 imshow 会将 (M,N,3) 形状的数组解释为RGB 数组。使用 interpolation='nearest' 对输出进行像素化:

import numpy as np
import matplotlib.pyplot as plt

# Example pixel array
pixels = np.random.rand(100,100,3)
reds = pixels[:,:,1:] = 0.          # Remove B and G

plt.imshow(reds, interpolation='nearest')

如果它存储为元组列表，则 np.array(...) 将执行转换。