Getting the indices of several elements in a NumPy array at once

by

You could use in1d and nonzero (or where for that matter):

>>> np.in1d(b, a).nonzero()[0]
array([0, 1, 4])

This works fine for your example arrays, but in general the array of returned indices does not honour the order of the values in a. This may be a problem depending on what you want to do next.

In that case, a much better answer is the one @Jaime gives here, using searchsorted:

>>> sorter = np.argsort(b)
>>> sorter[np.searchsorted(b, a, sorter=sorter)]
array([0, 1, 4])

This returns the indices for values as they appear in a. For instance:

a = np.array([1, 2, 4])
b = np.array([4, 2, 3, 1])

>>> sorter = np.argsort(b)
>>> sorter[np.searchsorted(b, a, sorter=sorter)]
array([3, 1, 0]) # the other method would return [0, 1, 3]

More Related Contents:

  1. Sorting arrays in NumPy by column
  2. How to access the ith column of a NumPy multidimensional array?
  3. Python/NumPy first occurrence of subarray
  4. Concatenating two one-dimensional NumPy arrays
  5. Numpy matrix to array
  6. Replacing Numpy elements if condition is met
  7. How to make a multidimension numpy array with a varying row size?
  8. How can I add new dimensions to a Numpy array?
  9. Force NumPy ndarray to take ownership of its memory in Cython
  10. How to normalize a NumPy array to within a certain range?
  11. How do I create an empty array and then append to it in NumPy?
  12. Immutable numpy array?
  13. How to get a list of all indices of repeated elements in a numpy array
  14. Binary random array with a specific proportion of ones?
  15. Compress numpy arrays efficiently
  16. Add multiple values to one numpy array index
  17. Index 2D numpy array by a 2D array of indices without loops
  18. Flattening a list of NumPy arrays?
  19. Element-wise string concatenation in numpy
  20. Fastest pairwise distance metric in python
  21. Partition array into N chunks with Numpy
  22. Numpy Vector (N,1) dimension -> (N,) dimension conversion
  23. numpy division with RuntimeWarning: invalid value encountered in double_scalars
  24. How does the axis parameter from NumPy work?
  25. numpy – evaluate function on a grid of points
  26. numpy arange: how to make “precise” array of floats?
  27. Shift rows of a numpy array independently
  28. Interweaving two NumPy arrays efficiently
  29. Formatting floats in a numpy array [duplicate]
  30. How to convert a 3d numpy array to 2d

Leave a Comment