配列Aの各要素について、配列Bにあるかないかをboolで返したい(in)

Question

前提・実現したいこと

配列Aに含まれる配列Bの要素を抽出したいと考えています。
イメージとしてはinを使うイメージでいましたが、うまくいきませんでした。

やりたいことは、AとBを使ってCを出力することです。

#やりたいこと
A = np.array([1,2,3,4,5])
B = np.array([1,3,5])
C = np.array([True, False, True, False, True])

#失敗
C = A in B
print(C)
#True

当然以下のようにforを使えばできるのですが、numpyやpandasでforはあまり使わない方が良いとも聞いているため、他に方法はありませんでしょうか。

C = []
for i in A:
    C.append(i in B)

Answer 1

numpy.vectorizeを使うとよいかと思います。
参考：numpy.vectorizeの使い方

A = np.array([1,2,3,4,5])
B = np.array([1,3,5])
vf = np.vectorize(lambda a: a in B)
C = vf(A)
print(C) #[ True False  True False  True]

Comments

[ataoka]

ありがとうございます！
希望通りの動作ができました。
迅速にご回答いただき助かりました。