原文:https://docs.scipy.org/doc/numpy/reference/generated/numpy.unique.html
校对:(虚位以待)
numpy.
unique
(ar, return_index=False, return_inverse=False, return_counts=False)[source]查找数组的唯一元素。
返回数组的排序的唯一元素。除了唯一元素外,还有三个可选输出:提供唯一值的输入数组的索引,重建输入数组的唯一数组的索引,以及输入数组中每个唯一值的出现次数。
参数: | ar:array_like
return_index:bool,可选
return_inverse:bool,可选
return_counts:bool,可选
|
---|---|
返回: | unique:ndarray
unique_indices:ndarray,可选
unique_inverse:ndarray,可选
unique_counts:ndarray,可选
|
也可以看看
numpy.lib.arraysetops
例子
>>> np.unique([1, 1, 2, 2, 3, 3])
array([1, 2, 3])
>>> a = np.array([[1, 1], [2, 3]])
>>> np.unique(a)
array([1, 2, 3])
返回给出唯一值的原数组的索引:
>>> a = np.array(['a', 'b', 'b', 'c', 'a'])
>>> u, indices = np.unique(a, return_index=True)
>>> u
array(['a', 'b', 'c'],
dtype='|S1')
>>> indices
array([0, 1, 3])
>>> a[indices]
array(['a', 'b', 'c'],
dtype='|S1')
从唯一值重新构建输入数组:
>>> a = np.array([1, 2, 6, 4, 2, 3, 2])
>>> u, indices = np.unique(a, return_inverse=True)
>>> u
array([1, 2, 3, 4, 6])
>>> indices
array([0, 1, 4, 3, 1, 2, 1])
>>> u[indices]
array([1, 2, 6, 4, 2, 3, 2])