numpyでindexを求めたい

Question

a=np.array(['a','b','c'])
b=np.array(['b','c','c','c'])
このとき
bと同じ形でaのインデックスを求めたいのですがどうしたら良いでしょうか？
求めたい配列=[1,2,2,2]

宜しくお願いします。

Answer 1

>>> np.where(b[:,None] == a)[1]
array([1, 2, 2, 2])

※ 今回の場合は a の要素が辞書順に並んでいるので以下でも可。

>>> np.searchsorted(a, b)
array([1, 2, 2, 2])

Comments

[a.‫com]

ありがとうございます！

[a.‫com]

ついでにお聞きしたいのですが
bが二次元だった場合は
b=[0:,0,None]と[0:,1,None]などのようにそれぞれwhereするしか無いのでしょうか？

[melian]

その場合は b を平坦化するとよいかと(最後に元の shape に戻します)。

np.where(b.reshape(-1,1)==a)[1].reshape(b.shape)

[a.‫com]

ありがとうございます！勉強になります。