Python Numpy中数组的集合操作详解_python_

s1 = {1, 2, 3} s2 = {2, 3, 4} """ &: 交集 |: 并集  -: 差集 ^: 对称差集 """ # 以下几种方式是等价的 # 但是一般我们都会使用操作符来进行处理，因为比较方便 print(s1 & s1) print(s1.intersection(s2)) print(set.intersection(s1, s2)) """ {2, 3} {2, 3} {2, 3} """ print(s1 | s2) print(s1.union(s2)) print(set.union(s1, s1)) """ {1, 2, 3, 4} {1, 2, 3, 4} {1, 2, 3, 4} """ print(s1 - s2, s2 - s1) print(s1.difference(s2),       s2.difference(s1)) print(set.difference(s1, s2),       set.difference(s2, s1)) """ {1} {4} {1} {4} {1} {4} """ print(s1 ^ s2) print(s1.symmetric_difference(s2)) print(set.symmetric_difference(s1, s2)) """ {1, 4} {1, 4} {1, 4} """ # 另外，我们还可以同时对多个集合操作，不仅仅是两个 print({1, 2, 3} & {2, 3, 4} & {3, 4, 5})  # {3}

import numpy as np arr1 = np.array([1, 2, 3]) arr2 = np.array([2, 3, 4]) # 两个数组 & # 表示将数组里面对应的元素分别进行"按位与"操作 print(arr1 & arr2)  # [0 2 0]

import numpy as np arr1 = np.array([1, 2, 2, 3]) arr2 = np.array([2, 3, 4, 4]) # 取交集 print(     np.intersect1d(arr1, arr2) )  # [2 3] # 取并集 print(     np.union1d(arr1, arr2) )  # [1 2 3 4] # 取差集 print(     np.setdiff1d(arr1, arr2),     np.setdiff1d(arr2, arr1) )  # [1] [4] # 取对称差集 print(     np.setxor1d(arr1, arr2) )  # [1 4]

from functools import reduce import numpy as np arr1 = np.array([1, 2, 3]) arr2 = np.array([2, 3, 4]) arr3 = np.array([3, 4, 5]) print(     reduce(np.intersect1d,            [arr1, arr2, arr3]) )  # [3]