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和 bool 索引:

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])