numpy.setdiff1d

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

Найти разность двух массивов.

Возвращает уникальные значения в ar1 , которые отсутствуют в ar2.

Параметры:

ar1array_like

Входной массив.

ar2array_like

Входной массив сравнения.

assume_uniquebool

Если True, предполагается, что оба входных массива уникальны, что может ускорить вычисление. По умолчанию False.

Возвращает:

setdiff1dndarray

Одномерный массив значений из ar1, которые отсутствуют в ar2. Результат отсортирован, когда assume_unique=False, в противном случае отсортирован только если входной массив был отсортирован.

Примеры

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