Numpy: For every element in one array, find the index in another array

by

I want to suggest one-line solution:

indices = np.where(np.in1d(x, y))[0]

The result is an array with indices for x array which corresponds to elements from y which were found in x.

One can use it without numpy.where if needs.

More Related Contents:

  1. Get the position of the largest value in a multi-dimensional NumPy array
  2. Generalise slicing operation in a NumPy array
  3. Numpy sub-array assignment with advanced, mixed indexing
  4. How is numpy’s fancy indexing implemented?
  5. Access n-th dimension in python [duplicate]
  6. Is there a NumPy function to return the first index of something in an array?
  7. Understanding NumPy’s einsum
  8. Intuition and idea behind reshaping 4D array to 2D array in NumPy
  9. Is there any numpy group by function?
  10. Computing the correlation coefficient between two multi-dimensional arrays
  11. Removing nan values from an array
  12. Why is numpy’s einsum faster than numpy’s built in functions?
  13. How to return 0 with divide by zero
  14. Reshape an array in NumPy
  15. Interweaving two numpy arrays
  16. Most efficient way to forward-fill NaN values in numpy array
  17. NumPy array initialization (fill with identical values)
  18. Why does numpy.zeros takes up little space
  19. Selecting Random Windows from Multidimensional Numpy Array Rows
  20. Searching a sequence in a NumPy array
  21. How to split a numpy array in fixed size chunks with and without overlap?
  22. Parameters to numpy’s fromfunction
  23. List to array conversion to use ravel() function
  24. Initialise numpy array of unknown length
  25. Efficient distance calculation between N points and a reference in numpy/scipy
  26. Index all *except* one item in python
  27. Why does numpy.linalg.solve() offer more precise matrix inversions than numpy.linalg.inv()?
  28. load csv into 2D matrix with numpy for plotting
  29. Convert list of lists with different lengths to a numpy array [duplicate]
  30. Select elements of numpy array via boolean mask array

Leave a Comment