python-numpy.unique在什么方面生成唯一列表？

内容导读

互联网集市收集整理的这篇技术教程文章主要介绍了python-numpy.unique在什么方面生成唯一列表？，小编现在分享给大家，供广大互联网技能从业者学习和参考。文章包含1469字，纯文字阅读大概需要3分钟。

内容图文

如果您向numpy.unique输入带有通用对象的数组,则结果将基于什么而唯一.

我努力了：

import numpy as np

class A(object): #probably exists a nice mixin for this :P
    def __init__(self, a):
        self.a = a
    def __lt__(self, other):
        return self.a < other.a
    def __le__(self, other):
        return self.a <= other.a
    def __eq__(self, other):
        return self.a == other.a
    def __ge__(self, other):
        return self.a >= other.a
    def __gt__(self, other):
        return self.a > other.a
    def __ne__(self, other):
        return self.a != other.a
    def __repr__(self):
        return "A({})".format(self.a)
    def __str__(self):
       return self.__repr__()

np.unique(map(A, range(3)+range(3)))

哪个返回

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

但我的意图是：

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

解决方法:

假设重复的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__.]

内容总结

以上是互联网集市为您收集整理的python-numpy.unique在什么方面生成唯一列表？全部内容，希望文章能够帮你解决python-numpy.unique在什么方面生成唯一列表？所遇到的程序开发问题。 如果觉得互联网集市技术教程内容还不错，欢迎将互联网集市网站推荐给程序员好友。

内容备注

版权声明：本文内容由互联网用户自发贡献，该文观点与技术仅代表作者本人。本站仅提供信息存储空间服务，不拥有所有权，不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容， 请发送邮件至 gblab@vip.qq.com 举报，一经查实，本站将立刻删除。

内容手机端