Numpy删除重复的行

Question

我只需要删除在数组中重复但保留其中一行的行,我不能使用unique,因为我需要维护顺序.例

1 a234 125
1 a123 265
1 a234 125
1 a145 167
1 a234 125    
2 a189 547
2 a189 547    
3 a678 567
3 a357 569

我需要这个输出

1 a234 125
1 a123 265
1 a145 167    
2 a189 547
3 a678 567
3 a357 569

Answer 1

我想这是你想要什么,并采用np.unique与return_index关键字参数:

import numpy as np

a = np.array([[1, 'a234', 125],
              [2, 'b189', 547],
              [1, 'a234', 125],
              [3, 'c678', 567],
              [1, 'a234', 125],
              [2, 'b189', 547]])

b = a.ravel().view(np.dtype((np.void, a.dtype.itemsize*a.shape[1])))
_, unique_idx = np.unique(b, return_index=True)

new_a = a[np.sort(unique_idx)]

>>> new_a
array([['1', 'a234', '125'],
       ['2', 'b189', '547'],
       ['3', 'c678', '567']], 
      dtype='|S4')

最讨厌的部分是视图b,它将每一行转换为单个元素np.void dtype,以便可以比较完整行的相等性np.unique.