numpy.unique在哪些方面生成一個唯一的列表？

Question

如果你輸入一個帶有一般對象的數組到numpy.unique，結果會根據什麼來唯一嗎？

我曾嘗試：

返回

array([A(0), A(0), A(1), A(1), A(2), A(2)], dtype=object) 

，但我的意圖是獲得：

array([A(0), A(1), A(2)], dtype=object) 

Answer 1

假設重複A(2)是一個錯字，我覺得你根本需要定義__hash__（請參閱docs）：

import numpy as np 
from functools import total_ordering 

@total_ordering 
class A(object): 
    def __init__(self, a): 
     self.a = a 
    def __lt__(self, other): 
     return self.a < other.a 
    def __eq__(self, other): 
     return self.a == other.a 
    def __ne__(self, other): 
     return self.a != other.a 
    def __hash__(self): 
     return hash(self.a) 
    def __repr__(self): 
     return "A({})".format(self.a) 
    def __str__(self): 
     return repr(self) 

產生

>>> map(A, range(3)+range(3)) 
[A(0), A(1), A(2), A(0), A(1), A(2)] 
>>> set(map(A, range(3)+range(3))) 
set([A(0), A(1), A(2)]) 
>>> np.unique(map(A, range(3)+range(3))) 
array([A(0), A(1), A(2)], dtype=object) 

，我已經使用total_ordering減少方法的擴散，你猜是可能的。 ：^）

[發佈更正缺少__ne__編後]