Get intersecting rows across two 2D numpy arrays

by

For short arrays, using sets is probably the clearest and most readable way to do it.

Another way is to use numpy.intersect1d. You’ll have to trick it into treating the rows as a single value, though… This makes things a bit less readable…

import numpy as np

A = np.array([[1,4],[2,5],[3,6]])
B = np.array([[1,4],[3,6],[7,8]])

nrows, ncols = A.shape
dtype={'names':['f{}'.format(i) for i in range(ncols)],
       'formats':ncols * [A.dtype]}

C = np.intersect1d(A.view(dtype), B.view(dtype))

# This last bit is optional if you're okay with "C" being a structured array...
C = C.view(A.dtype).reshape(-1, ncols)

For large arrays, this should be considerably faster than using sets.

More Related Contents:

  1. Numpy “where” with multiple conditions
  2. “Cloning” row or column vectors
  3. efficiently convert uneven list of lists to minimal containing array padded with nan
  4. How does NumPy’s transpose() method permute the axes of an array?
  5. How to normalize a NumPy array to a unit vector?
  6. Efficient Numpy 2D array construction from 1D array
  7. Replacing Pandas or Numpy Nan with a None to use with MysqlDB
  8. Swap slices of Numpy arrays
  9. Installing numpy with pip on windows 10 for python 3.7
  10. Why does pyplot.contour() require Z to be a 2D array?
  11. Cython: “fatal error: numpy/arrayobject.h: No such file or directory”
  12. How do I stack vectors of different lengths in NumPy?
  13. Cumsum reset at NaN
  14. Finding index of nearest point in numpy arrays of x and y coordinates
  15. Multiplying across in a numpy array
  16. initialize a numpy array
  17. Numpy meshgrid in 3D
  18. Sending live video frame over network in python opencv
  19. What is the difference between native int type and the numpy.int types?
  20. What does dtype=object mean while creating a numpy array?
  21. Performance: Matlab vs Python
  22. Selecting close matches from one array based on another reference array
  23. Efficiently sorting a numpy array in descending order?
  24. Numpy: Divide each row by a vector element
  25. Running maximum of numpy array values
  26. How to get the cumulative distribution function with NumPy?
  27. Load CSV to Pandas MultiIndex DataFrame
  28. Python pandas – new column’s value if the item is in the list
  29. Digitizing an analog signal
  30. calculate distance of 2 list of points in numpy

Leave a Comment