numpy.setdiff1d

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

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

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

Parameters

ar1array_like

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

ar2array_like

Массив для сравнения.

assume_uniquebool

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

Returns

setdiff1dndarray

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

См. также

numpy.lib.arraysetops

Модуль с рядом других функций для выполнения операций над массивами по принципу множеств.

Примеры

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