python - 从该列的值检索二维 numpy 数组中的列索引

Question

如果我有两个 numpy 数组，例如:

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

我想获取与b的值相对应的a列的索引。这里是 0。

类似:

np.where(np.hsplit(a, 4) == b)

我能够找到解决方案，但我认为这应该是一些更直观的方法。

Answer 1

看看这个 answer 。 唯一的区别是您需要转置a。

>>> a = np.array([[ 0,  1,  2,  3],
              [ 4,  5,  6,  7],
              [ 8,  9, 10, 11]])
>>> b = np.array([ 0, 4, 8])
>>> np.all(a.T==b,axis=1)
array([ True, False, False, False])
>>> np.where(np.all(a.T==b,axis=1))[0][0]
0