【备份】numpy.unique()

import numpy as np

arr = np.array([1, 2, 2, 3, 3, 3, 4, 4, 4, 4])

# 返回一个值
unique_elements = np.unique(arr)
print("唯一元素：", unique_elements)  # 从小到大排序

# 返回两个值
# 1、提取唯一元素和索引：元素在原始数组中第一次出现的索引
unique_elements, indices = np.unique(arr, return_index=True)
print("唯一元素：", unique_elements)
print("第一次出现的索引：", indices)

# 2、提取唯一元素和反向索引：原始数组中每个元素在唯一元素数组中的索引
unique_elements, inverse_indices = np.unique(arr, return_inverse=True)
print("唯一元素：", unique_elements)
print("反向索引：", inverse_indices)

# 3、提取唯一元素和元素出现次数
unique_elements, counts = np.unique(arr, return_counts=True)
print("唯一元素：", unique_elements)
print("出现次数：", counts)

输出：

唯一元素： [1 2 3 4]
唯一元素： [1 2 3 4]
第一次出现的索引： [0 1 3 6]
唯一元素： [1 2 3 4]
反向索引： [0 1 1 2 2 2 3 3 3 3]
唯一元素： [1 2 3 4]
出现次数： [1 2 3 4]