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, но в противном случае отсортирован только в том случае, если входной массив отсортирован.

См. также

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])