當前位置:成語大全網 - 新華字典 - 如何使用python將二維數組去重呢

如何使用python將二維數組去重呢

方案1:轉化為虛數

x=c[:,0]+c[:,1]*1j

print('轉化為虛數:',x)

print('虛數去重後:',np.unique(x))

print(np.unique(x,return_index=True))#return_index:輸出的元素索引值

idx=np.unique(x,return_index=True)[1]

print('二維數組去重:\n',c[idx])

#方案2:利用set

print('去重方案2:\n',np.array(list(set([tuple(t) for t in c]))))