将 2D 数组(灰度图像)转换为 3D(RGB 图像)数组

Question

我有一个灰度图像作为 2d numpy 数组。我想将其转换为 RGB 图像作为 3d numpy 数组。

RGB 颜色是随机生成的。

import numpy as np
data = np.random.randint(low=0, high=255, size=(25, 25))
uniq_data = np.unique(data)
print (uniq_data)
colors = np.random.randint(low=0, high=255, size=(len(uniq_data), 3))
result = ?

我不知道该怎么做。

我需要为唯一数据放置唯一颜色(RGB 值(。 输出形状为 (25， 25， 3(。

Answer 1

制作一个查找表并通过高级索引使用它：

lookup = np.zeros((255,3),dtype=np.uint8)
lookup[uniq_data] = colors
rgb = lookup[data]