Chciałbym użyć funkcji numpy.where w tablicy ciągów. Nie udaje mi się to jednak zrobić. Czy ktoś może mi pomóc to rozgryźć?

Na przykład, kiedy używamnumpy.where w następującym przykładzie pojawia się błąd:

import numpy as np

A = ['apple', 'orange', 'apple', 'banana']

arr_index = np.where(A == 'apple',1,0)

Otrzymuję następujące informacje:

>>> arr_index
array(0)
>>> print A[arr_index]
>>> apple

Chciałbym jednak znać indeksy w tablicy ciągów,A gdzie sznurek'apple' mecze. W powyższym ciągu dzieje się to w 0 i 2. Jednaknp.where zwraca tylko 0, a nie 2.

Więc jak mam to zrobićnumpy.where pracować na strunach? Z góry dziękuję.