for ループを使わないnumpyの要素置換方法が知りたい！

Question

前提・実現したいこと

for ループを使用せずに、条件を満たす要素の置換を行いたい。

発生している問題・エラーメッセージ

数千、数万以上のループによって置換が必要となるため
高速にできる手法があれば知りたい。

3次元配列aについて、bの要素と一致するものを100に置換したい。
簡素化のため、下記ソースコードは小さくしてあります。

該当のソースコード

　a = np.array([[[0,1,2],[3,4,5]],[[6,7,8],[9,10,11]]])
  b = np.array([1,5,9])
    
  for i in range(len(b)):
        x = b[i]
        a = np.where(a == x, 100, a)

  print(a)

試したこと

上記のようなコード作成したが、ループ数が多いと時間がかかりすぎる。

補足情報（FW/ツールのバージョンなど）

Python3

Answer 1

isinを使うとbのどれかと一致するaのインデックスを抽出できます。

import numpy as np

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

a[np.isin(a, b)] = 100

print(a)

Comments

[chanchi]

ありがとうございました。助かりました！