Python numpy setdiff1d用法及代碼示例

本文簡要介紹 python 語言中 numpy.setdiff1d 的用法。

用法:

numpy.setdiff1d(ar1, ar2, assume_unique=False)

求兩個數組的集差。

返回中的唯一值ar1那些不在ar2.

參數：

ar1： array_like

輸入數組。

ar2： array_like

輸入比較數組。

assume_unique： bool

如果為 True，則假定輸入數組都是唯一的，這樣可以加快計算速度。默認為假。

返回：

setdiff1d： ndarray

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