numpy.setdiff1d

numpy.setdiff1d(ar1, ar2, assume_unique=False) [source]

Find the set difference of two arrays.

Return the sorted, unique values in ar1 that are not in ar2.

Parameters:
ar1 : array_like

Input array.

ar2 : array_like

Input comparison array.

assume_unique : bool

If True, the input arrays are both assumed to be unique, which can speed up the calculation. Default is False.

Returns:
setdiff1d : ndarray

Sorted 1D array of values in ar1 that are not in ar2.

See also

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

Examples

>>> a = np.array([1, 2, 3, 2, 4, 1])
>>> b = np.array([3, 4, 5, 6])
>>> np.setdiff1d(a, b)
array([1, 2])

© 2005–2019 NumPy Developers
Licensed under the 3-clause BSD License.
https://docs.scipy.org/doc/numpy-1.15.4/reference/generated/numpy.setdiff1d.html