使用numpy.ndarray属性初始化类

Question

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

该脚本将按预期方式对一维阵列运行。 但是，在多维数组的情况下，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的初始化过程，从而产生您所看到的错误。

有关更多信息，请参见有关对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