numpy 数组元素去重与数组元素数量统计 np.unique()

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

Params

• return_index：默认False。设置成True时，返回去重后数组元素在传入数组中第一次出现的index
• return_inverse：默认False。设置成True时，返回传入数组中的元素在去重数组中的index，用来重构传入数组。所以，叫inverse，即反向生成传入数组。
• return_counts：默认False。设置成True时，返回传入数组中元素重复出现的次数。
• axis：默认None，此时会将传入数组flattern成一维数组后再进行unique操作。根据传入数组维度修改后，会在指定axis上统计唯一元素。

Return：
默认参数下，返回一个去重的、升序排序的一维数组

举例说明：
例1：

arr = np.array([[2,7,9],[2,3,8]])
np.unique(arr,return_index=True,return_counts=True,return_inverse=True)
------------------------------------------------
(array([2, 3, 7, 8, 9]),
 array([0, 4, 1, 5, 2], dtype=int64),
 array([0, 2, 4, 0, 1, 3], dtype=int64),
 array([2, 1, 1, 1, 1], dtype=int64))

返回一个tuple，因为axis默认是None，这里我们已经flattern了，是统计的[2,7,9,2,3,8]这个数组。

• 第一个array时排序后的唯一元素。
• 第二个array依次是唯一元素2,3,7,8,9第一次出现的index。
• 第三个是反向index，即[2,7,9,2,3,8]这个传入数组中的元素在唯一元素数组[2, 3, 7, 8, 9]中的index。
• 第四个是统计重复数量数组。

当设置axis=0或者1时，则是在另一个轴上进行元素的排序，当两个元素第一个数相同时，就会按照第二个数进行排序。