[SciPy-User] equivalent of tolist().index(entry) for numpy 1d array of strings
Ryan Krauss
ryanlists at gmail.com
Mon Dec 21 20:53:19 EST 2009
I wrote some code to work with csv spreadsheet files by reading the
columns into lists, but I need to rework the code to work with numpy
1d arrays of strings rather than lists. I need to search one of these
columns/arrays. What is the best way to find the index for the
element that matches a certain string (or maybe just the first element
to match such a string)?
With the columns as lists, I was doing
index = mylist.index(entry)
So, I could obviously do
index = mylist.tolist().index(entry)
but I don't know if that would be slower or clumsier than something like
bool_vect = where(mylist==entry)[0]
index = bool_vect[0]
or just
index = where(mylist==entry)[0][0]
Any thoughts? Is there an easier way?
Thanks,
Ryan
More information about the SciPy-User
mailing list