numpy.setdiff1d#

numpy.setdiff1d(ar1, ar2, assume_unique=False)[источник]#

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

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

Параметры:

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

Возвращает:

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

Примеры

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