唯一值与其他逻辑集合
2019-01-30 本文已影响6人
庵下桃花仙
对一维 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页)