索引中给定一个查找表numpy的阵列

Question

：

colors = [ [0,0,0],\ 
      [0, 255, 0],\ 
      [0, 0, 255],\ 
      [255, 0, 0]] 

和索引numpy的矩阵输入2×2：

a = np.array([[0,1],[1,1]]) 

我如何映射到一个2x2x3矩阵b，其中b[i][j] = colors[a[i][j]]？我想避免在这里使用for循环。

Answer 1

你试过：

colors[a] 

这里有一个完整的例子：

import numpy as np 

colors = np.array([[0,0,0], 
        [0, 255, 0], 
        [0, 0, 255], 
        [255, 0, 0] 
        ]) 
a = np.array([[0, 1], [1, 1]]) 

new = colors[a] 
new.shape 
# (2, 2, 3) 
new 
# array([[[ 0, 0, 0], 
#   [ 0, 255, 0]], 
# 
#  [[ 0, 255, 0], 
#   [ 0, 255, 0]]])