numpy 数组集合运算及下标操作
1. 数组的集合运算 1.1. 并集 np.union1d(a,b)计算数组的并集: In [1]: import numpy as np In [2]: a = np.array([1,2,3]) In [3]: b = np.array([3,4,5]) In [4]: np.union1d(a,b) Out[4]: array([1, 2, 3, 4, 5]) 1.2. 交集 np.intersect1d(a,b)计算数组的交集: In [10]: import numpy as np In [11]: a = np.array([2,3,4,5]) In [12]: b = np.array([3,5,6]) In [13]: np.intersect1d(a,b) Out[13]: array([3, 5]) 1.3. 差集 np.setdiff1d(a,b)计算结果为a集合减去b集合,也就是剩下没有在b中出现的元素: In [15]: import numpy as np In [16]: a = [1,1,2,2,3,4,5] In [17]: b = [2,5] In [18]: np.setdiff1d(a,b) Out[18]: array([1, 3, 4]) 1.4. 异或 np.setxor1d(a,b)计算a、b集合的异或结果。 In [15]: