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