python代码 二值向量不会写 numpy

Question

请问变成二值向量应该如何写这个代码？随便一个向量，希望有普遍性

Answer 1

import numpy as np

a = np.array([1, 2, 3, 4, 5, 6, 7])
b = np.array([3, 6])

c = np.array(list(map(lambda x: int(x in b) ,a)))
print(c)