我有一个形状为(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]]