【Numpy】np.unique去重复

本文主要是介绍【Numpy】np.unique去重复，希望对大家解决编程问题提供一定的参考价值，需要的开发者们随着小编来一起学习吧！

 

numpy.unique(ar, return_index=False, return_inverse=False, return_counts=False, axis=None)

去除重复，返回索引值，排序

>>> 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([[1, 0, 0], [1, 0, 0], [2, 3, 4]])
>>> np.unique(a, axis=0)
array([[1, 0, 0], [2, 3, 4]])

return_index将得到结果对应于原始数组中的索引值。 

>>> a = np.array(['a', 'b', 'b', 'c', 'a'])
>>> u, indices = np.unique(a, return_index=True)
>>> u
array(['a', 'b', 'c'], dtype='<U1')
>>> indices
array([0, 1, 3])
>>> a[indices]
array(['a', 'b', 'c'], dtype='<U1')

return_inverse记录恢复原状所需要的索引值 

>>> 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, ..., 1, 2, 1])
>>> u[indices]
array([1, 2, 6, ..., 2, 3, 2])

 

这篇关于【Numpy】np.unique去重复的文章就介绍到这儿，希望我们推荐的文章对编程师们有所帮助！

---