自動調整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__調用中省略了*和**運算符。