NumPy 中的集合運算

怎樣快速找出兩個數組中相同的元素？

numpy.isin（element，test_elements，assume_unique = False，invert = False ）
計算test_elements中的元素，僅在元素上廣播。返回與元素相同形狀的布爾數組，該元素為True，其中元素元素位於test_elements中，否則返回 False。

 NumPy數組的集合運算

import numpy as np
# 創建一維 ndarray x
x = np.array([1,2,3,4,5])
# 創建一維 ndarray y
y = np.array([6,5,4,8,7])
# 輸出 x
print('x = ', x)
# 輸出 y
print('y = ', y)
# 比較x和y
print('使用intersect1d輸出x和y的交集:', np.intersect1d(x,y))
print('使用setdiff1d輸出在x中不在y中的元素:', np.setdiff1d(x,y))
print('使用union1d輸出x和y的並集:',np.union1d(x,y))
print('使用setxor1d輸出x和y的異或集:',np.setxor1d(x,y))

輸出：

x =  [1 2 3 4 5]
y =  [6 5 4 8 7]
使用intersect1d輸出x和y的交集: [4 5]
使用setdiff1d輸出在x中不在y中的元素: [1 2 3]
使用union1d輸出x和y的並集: [1 2 3 4 5 6 7 8]
使用setxor1d輸出x和y的異或集: [1 2 3 6 7 8]