numpy学习笔记-唯一化

numpy提供了一些针对一维ndarray的基本集合运算。最常用的可能要数np.unique了，他用于找出数组中唯一值并返回已排序的结果。

names = np.array(['Bob','Joe','Will','Bob','Will','Joe','Joe'])
print np.unique(names)

输出的结果
['Bob' 'Joe' 'Will']

ints = np.array([3,3,3,2,2,1,1,4,4])
print np.unique(ints)

输出的结果
[1 2 3 4]

与上面等价的纯python代码如下

sorted(set(names))

另一个函数np.in1d用于测试一个数组中的值在另一个数组中的成员资格，返回一个布尔型数组（即检查一个数组中的值是否在另一个数组中出现）：

values = np.array([6,0,0,3,2,5,6])
print np.in1d(values,[2,3,6])

结果
[ True False False  True  True False  True]