自動調整大小的NumPy recarray

Question

我想創建一個numpy.recarray的子​​類，當數據添加到當前長度以外的行時，它會自動調整大小。

下面的代碼可以完成我想要的大部分功能。

class autorecarray(numpy.recarray): 

    def __init__(self,*args,**kwargs): 
     self._increment = 1 
     numpy.recarray.__init__(self,args,kwargs) 

    def __setitem__(self,ind,y): 
     try: 
     numpy.recarray.__setitem__(self,ind,y) 
     except IndexError: 
     self.resize((self.__len__()+self._increment,),refcheck=False) 
     self.__setitem__(ind,y) 

它工作正常的這種使用情況：

a = utils.autorecarray((1,),formats=['i4','i4']) 
a[1] = (1,2) # len(a) will now be 2 

然而，這種用法將提高對numpy.core.records.recarray __getitem__方法的IndexError：

a[2]['f1'] = 3 

我最初的嘗試還要覆蓋我的子類中的__getitem__方法，但此代碼不起作用。

def __getitem__(self,ind): 
     try: 
     numpy.recarray.__getitem__(self,ind) 
     except IndexError: 
     self.resize((self.__len__() + self._increment,),refcheck=False) 
     self.__getitem__(ind) 

它不會自動擴展陣列，但現在的陣列中的每個產品None，不能改變。

誰能告訴我我做錯了什麼？

Answer 1

首先你錯過了在numpy.recarray.__init__通話的星號：

def __init__(self, *args, **kwargs): 
    self._increment = 1 
    numpy.recarray.__init__(self, *args, **kwargs) 

第二，你錯過了在__getitem__return聲明：

def __getitem__(self,ind): 
    try: 
     return numpy.recarray.__getitem__(self,ind) 
    except IndexError: 
     self.resize((self.__len__() + self._increment,),refcheck=False) 
     return self.__getitem__(ind) 

Answer 2

重寫的__getitem__不返回值。

我花了很長時間才意識到這一點。

此外，如Petr Viktorin指出，您在__init__調用中省略了*和**運營商。