使用初始化类numpy.ndarray公司属性

Question

我正在尝试定义一个具有numpy.ndarray类型属性的包装类。必须通过调用__init__()初始化该属性。你知道吗

对于1D数组，脚本按预期运行。但是，对于多维数组，python返回以下错误：only length-1 arrays can be converted to Python scalars

import numpy as np

class myArr(np.ndarray):
    def __init__(self,Arr):
        self.Arr = Arr

npArr = np.zeros((3)) # works
#npArr = np.zeros((3,5)) # does not work
print npArr

wrappedArr = myArr(npArr)
print wrappedArr.Arr

这里发生了什么？你知道吗

python 2.7.6，numpy 1.8.2

Answer 1

如果您只想拥有一个类型为ndarray的属性，您从中继承它有什么具体的原因吗？你知道吗

我想说，通过子类化ndarray和重写__init__，您将破坏numpy的初始化过程，从而产生您看到的错误。你知道吗

有关详细信息，请参阅关于Subclassing ndarray的numpy文档。你知道吗

从object继承（将MyArr转换为常规的新样式类）可以解决您的问题：

import numpy as np

class MyArr(object):
    def __init__(self, arr):
        self.arr = arr

np_arr_1 = np.zeros((3))
np_arr_2 = np.zeros((3, 5))

wrapped_arr_1 = MyArr(np_arr_1)
wrapped_arr_2 = MyArr(np_arr_2)

print wrapped_arr_1.arr
print wrapped_arr_2.arr