Python：使用数组作为另一个数组的条件

Question

是否可以简单地使用类似以下的数组：

import numpy as np
a = np.array([1,76,120,158,199]

作为另一个数组的条件数组？

我有例如一个数组：

b=np.array([0,1,2,3,4,5,76,77,78,79,120,121,122,158,159,199,200,201])

数组b中所有也在数组a中的所有数字都应为1，例如：

返回：

c=np.array([0,1,2,3,4,5,1,77,78,79,1,121,122,1,159,1,200,201])

所以有一个快速的numpy版本可以做到这一点。 因为实际的数组是一个巨大的3d数组，具有许多数字，所以可以将其展平。

Answer 1

使用numpy.in1d和布尔索引：

b[np.in1d(b, a)] = 1

print(b)

array([  0,   1,   2,   3,   4,   5,   1,  77,  78,  79,   1, 121, 122,
         1, 159,   1, 200, 201])