Numpy " où " sur la chaîne

Je voudrais utiliser le numpy.où la fonction sur un tableau de chaînes. Cependant, je suis échoué à le faire. Quelqu'un peut-il m'aider à comprendre cela?

Par exemple, lorsque j'utilise numpy.where sur l'exemple suivant, j'obtiens une erreur:

import numpy as np

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

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

Je reçois le texte suivant:

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

Cependant, je voudrais savoir les indices dans le tableau de chaîne, A où la chaîne 'apple' matchs. Au-dessus de la chaîne de ce qui se passe à 0 et 2. Cependant, la np.where ne renvoie 0 et pas 2.

Alors, comment dois-je faire numpy.where travaux sur cordes? Merci à l'avance.

OriginalL'auteur Rohit | 2013-05-20

  1. 8
    print a[arr_index]

    pas array_index!!

    a = np.array(['apple', 'orange', 'apple', 'banana'])

arr_index = np.where(a == 'apple')

print arr_index

print a[arr_index]
    Merci pour la rapide réponse! Ah, c'est correct. Oui, mais ce que je veux sont les indices où la chaîne de "pomme" correspond dans le tableau de chaînes. Donc, je veux quelque chose comme: arr_index = [0,2] # c'est "pomme" se sont les positions 0 et 2 dans le tableau de chaînes un. Cependant, il ne me donne 0.
    J'ai juste corrigé l'erreur dans mon post de sorte que la question principale est mise en surbrillance et pas la stupide erreur. 🙂
    regarder le modifier j'ai fait et essayer de l'exécuter
    Grâce Joran, il a travaillé.

    OriginalL'auteur Joran Beasley