Question

我有一个来自numpy.ndarray的子类，它修改了get-item-by-index行为。但是，当有人在我的子类上打印时，我想保留原始的ndarray输出格式。我尝试重写__iter__函数，但似乎没有用。如果__getitem__超载，是否可以在底层数组顺序中输出打印数字？

import numpy as np

class imarray(np.ndarray):
    def __new__(cls, d, f):
        nd = np.asarray(d)
        return super(imarray, cls).__new__(cls,(nd.size,), dtype=nd.dtype, buffer=nd).view(imarray)
    def __init__(self, d, f):
        self.f = f
    def __array_finalize__(self, o):
        pass
    def __getitem__(self, i):
        return super(imarray, self).__getitem__(self.f(i))
    def __setitem__(self, i, y):
        return super(imarray, self).__setitem__(self.f(i), y)
    def __iter__(self):
        return super(imarray, self).__iter__

if __name__ == "__main__":
    a = imarray([1., 2., 3.1415], lambda i:i+1)
    print a[0] # print .2
    a[-1] =.4
    print a[-1] # print .4
    print a # print [ 2. 3.1415 0.4 ] but want [ 0.4 2. 3.1415 ]

Answer 1

Print返回对象__str__方法。

所以你需要覆盖__str__。

>>> class ex():
    def __init__(self, value):
        self.val = value
    def __str__(self):
        return self.val * 2


>>> a = ex('A')
>>> print a
AA

Answer 2

实际上取决于。

如果您打印对象，print会将其称为__str__方法。

但是，如果您打印list（或set或dict），print会调用列表的__str__方法，而后者又会调用项目的 __repr__ 方法。这很奇怪，但很容易观察到：

>>> class B():
...     def __repr__(self): return '__repr__'
...     def __str__(self): return '__str__'
...
>>> print B()
__str__
>>> print [B()]
[__repr__]
>>> print set([B()])
set([__repr__])
>>> print {B(): B()}
{__repr__: __repr__}

我应该重载什么功能来定制打印输出？

2 个答案: