学科分类
目录
数据分析

唯一化及其他集合逻辑

针对一维数组,NumPy提供了unique()函数来找出数组中的唯一值,并返回排序后的结果,示例代码如下。

In [124]: arr = np.array([12, 11, 34, 23, 12, 8, 11])
In [125]: np.unique(arr)
Out[125]: array([ 8, 11, 12, 23, 34])

除此之外,还有一个in1d()函数用于判断数组中的元素是否在另一个数组中存在,该函数返回的是一个布尔型的数组,示例代码如下。

In [126]: np.in1d(arr, [11, 12])
Out[126]: array([ True, True, False, False, True, False, True])

NumPy提供的有关集合的函数还有很多,表1列举了数组集合运算常见的函数。

表1 数组集合运算的常见函数

函数 描述
unique(x) 计算x中的唯一元素,并返回有序结果
intersect1d(x,y) 计算x和y中的公共元素,并返回有序结果
union1d(x,y) 计算x和y的并集,并返回有序结果
in1d(x,y) 得到一个表示“x的元素是否包含y”的布尔型数组
setdiff1d(x,y) 集合的差,即元素在x中且不在y中
setxor1d(x,y) 集合的对称差,即存在于一个数组中但不同时存在于两个数组中的元素
点击此处
隐藏目录