[Numpy-discussion] Distance Formula on an Array

Eric Firing efiring at hawaii.edu
Sun Apr 26 16:49:24 EDT 2009


Ian Mallett wrote:
> It would be:
> numpy.where(array<d)
> right?

Almost.  where(cond) is equivalent to nonzero(cond), and both return 
tuples.  Assuming your array is 1-D, you can use:

numpy.where(array<d)[0]
numpy.nonzero(array<d)[0]
numpy.flatnonzero(array<d)

But do you really need the indices?  If what you need is values of some 
array at those locations, then index directly with the condition:

A_subset = A[array<d]

Eric



More information about the NumPy-Discussion mailing list