用zip
当用np.unique给list去重后,可以返回提供的元素索引,但是会自动排序。这时候有时候会想要按照原来顺序(即索引顺序)排列去重结果。
np.unique用法,参考官网文档:
https://docs.scipy.org/doc/numpy-1.13.0/reference/generated/numpy.unique.html
zip用法:zip可以组合list,也可以拆分list。
category_uniques, index = np.unique(category,return_index=True)
temp = zip(index,category_uniques)
temp = sorted(temp)
index, category_uniques = zip(*temp)
或许你疑问为什么这么麻烦,直接用判断a in list 就好,但是,遇到非一维的list,就不好用了。