[Numpy-discussion] match or vectorized in-type function.

Zachary Pincus zachary.pincus@yale....
Mon Jul 12 21:32:16 CDT 2010


> match(v1, v2) => returns a boolean array of length len(v1) indicating
> whether element i in v1 is in v2.

You want numpy.in1d (and friends, probably, like numpy.unique and the  
others that are all collected in numpy.lib.arraysetops...)


Definition:       numpy.in1d(ar1, ar2, assume_unique=False)
Docstring:
     Test whether each element of a 1D array is also present in a  
second array.

     Returns a boolean array the same length as `ar1` that is True
     where an element of `ar1` is in `ar2` and False otherwise.

     Parameters
     ----------
     ar1 : array_like, shape (M,)
         Input array.
     ar2 : array_like
         The values against which to test each value of `ar1`.
     assume_unique : bool, optional
         If True, the input arrays are both assumed to be unique, which
         can speed up the calculation.  Default is False.

     Returns
     -------
     mask : ndarray of bools, shape(M,)
         The values `ar1[mask]` are in `ar2`.

     See Also
     --------
     numpy.lib.arraysetops : Module with a number of other functions for
                             performing set operations on arrays.



More information about the NumPy-Discussion mailing list