对一维 ndarry 的基础集合操作。如 np.unique 返回的是数组中唯一值排序后形成的数组。

In [123]: names = np.array(['bob', 'joe', 'will', 'bob', 'will', 'joe', 'joe'])

In [124]: np.unique(names)
Out[124]: array(['bob', 'joe', 'will'], dtype='<U4')

In [125]: ints = np.array([3, 3, 3, 2, 2, 1, 1, 4, 4])

In [126]: np.unique(ints)
Out[126]: array([1, 2, 3, 4])

np.unique和纯Python比较

In [127]: sorted(set(names))
Out[127]: ['bob', 'joe', 'will']

np.in1d 检查一个数组中的值是否存在于另一个数组中。

In [128]: values = np.array([6, 0, 0, 3, 2, 5, 6])

In [129]: np.in1d(values, [2, 3, 6])
Out[129]: array([ True, False, False,  True,  True, False,  True])

(《利用Python进行数据分析》115页)