如何初始化numpy.ndarray作为基类

时间:2013-03-02 12:06:56

标签: python numpy

我想将numpy.ndarray子类化为自定义某些行为。作为第一步,我尝试创建我自己的类,如下所示。但我还没有得到正确的初始化。下面我用[1.,2。]初学我的课,并期望“打印一行”这一行会让我[1.,2。]。但它总是返回我[[some_random_number,some_random_number]]。因此尺寸错误,初始值的分配失败。请帮忙。我无法理解。

import numpy as np

class imarray(np.ndarray):
  def __init__(self, d):
      data = np.asarray(d)
      super(imarray, self).__init__((data.size,), dtype=data.dtype, buffer=data)

if __name__ == "__main__":
  a = imarray([1.,2.])
  print a

0 个答案:

没有答案