将2d Numpy数组索引到更大的2d Numpy数组

时间:2019-04-05 22:05:55

标签: python arrays numpy

我有一个形状为(n,4)的numpy数组,我想用它索引到一个形状为(n,8)的大型numpy 2d数组。值在每行中最多出现一次,并且所有值都不大于7。我想做的就是获取每一行的关联列。

例如

a = np.array([[0, 2, 4, 6],
       [0, 3, 5, 6]])

b = np.array([[0, 1, 2, 3, 4, 5, 6, 7],
       [8, 9, 10, 11, 12, 13, 14, 15]])

b[a] should result in 

[[0, 2, 4, 6], [8, 11, 13, 14]]

0 个答案:

没有答案