逐行索引NumPy 2D阵列由另一个2D阵列

时间:2016-12-04 05:24:54

标签: python python-2.7 numpy indexing scipy

我有一个二维数组,如:

a = [[0, 0, 0],
     [1, 1, 1],
     [2, 2, 2],
     [3, 3, 3]]

我的索引数组如下:

b = [[0,0,1],
     [2,1,3]]

我的目标数组应该是:

c = [[[0,0,0],
      [0,0,0],
      [1,1,1]],    
     [[2,2,2],
      [1,1,1],
      [3,3,3]]]

如何使用NumPy包?是否有任何内置函数可以做到这一点? 谢谢你的帮助。

0 个答案:

没有答案