Python：使用numpy.unique() 统计list或array中所包含的元素，即返回不重复元素

import numpy as np
# a = [1,2,3,4,5,6,1,2,3,4,5,6,1,2,3,4,5,6,6]
a = np.array([1,2,3,4,5,6,1,2,3,4,5,6,1,2,3,4,5,6,6])
b,s ,t,w= np.unique(a,return_counts=True,return_index=True,return_inverse=True)
print(b)
print(s)
print(t)
print(w)

[1 2 3 4 5 6]
[0 1 2 3 4 5]
[0 1 2 3 4 5 0 1 2 3 4 5 0 1 2 3 4 5 5]
[3 3 3 3 3 4]

用于统计数据集类别数和各类别样本个数非常方便。