其他分享
首页 > 其他分享> > 去除array重复的元素

去除array重复的元素

作者:互联网

转自:https://blog.csdn.net/weixin_39858245/article/details/110776936

1.例子

import numpy as np

a=np.array([[1, 1, 1, 0, 0, 0],
[0, 1, 1, 1, 0, 0],
[0, 1, 1, 1, 0, 0],
[1, 1, 1, 0, 0, 0],
[1, 1, 1, 1, 1, 0]])
unique_rows =np.unique(a, axis=0)
reture_countsarr, uniq_cnt = np.unique(a, axis=0, return_counts=True)
original_array= reture_countsarr[uniq_cnt==1]

输出:

>>> unique_rows
         
array([[0, 1, 1, 1, 0, 0],
       [1, 1, 1, 0, 0, 0],
       [1, 1, 1, 1, 1, 0]])

>>> uniq_cnt
         
array([2, 2, 1], dtype=int64)

>>> reture_countsarr
         
array([[0, 1, 1, 1, 0, 0],
       [1, 1, 1, 0, 0, 0],
       [1, 1, 1, 1, 1, 0]])

>>> original_array
         
array([[1, 1, 1, 1, 1, 0]])

2.np.unique() 

import numpy as np

A = [1, 2, 2, 5,3, 4, 3]
a, s= np.unique(A, return_index=True)
print(a)
print(s)

#输出:
[1 2 3 4 5]
[0 1 4 5 3]

返回在原数组的下标。

标签:cnt,uniq,重复,reture,去除,np,array,unique
来源: https://www.cnblogs.com/BlueBlueSea/p/15853843.html