Question

我正在尝试对numpy数组进行子类化，以创建具有周期性边界条件的数组。我只需要使用它来处理一维数组。

我想要的：假设我有一个对象a=Periodic_Grid([0,1,2,3,4,5,6,7,8,9])。我想要以下输出

>>> a=Periodic_Grid(a)
>>> a
Periodic_Grid([0, 1, 2, 3, 4, 5, 6, 7, 8, 9])
>>> a[0]
0
>>> a[7]
7
>>> a[14]
4
>>> a[-5]
5
>>> a[-14]
6
>>> a[1:5]
array([1., 2., 3., 4.])
>>> a[8:13]
array([8., 9., 0., 1., 2.])
>>> a[-5:2]
array([5., 6., 7., 8., 9., 0., 1.])

换句话说，索引只是环绕。切片也应适当包裹。否则，它会正常工作。

最低工作代码

class Periodic_Grid(np.ndarray):
    def __new__(cls, input_array):
        obj = np.asarray(input_array).view(cls)
        obj.grid_size = input_array.size
        return obj 

    def __getitem__(self, index):
        if isinstance(index, slice):
            indicies=list(range(index.start, index.stop))
            size=len(indicies)
            a=np.empty(size)
            print("Size: {0} indicies: {1}".format(size, indicies))
            for i, ind in enumerate(indicies): 
                idx=self.wrapped_index(ind)
                a[i]=self[idx]
            return a

        elif isinstance(index, tuple):
            idx=index[0]
            idx=self.wrapped_index(idx)
            return super().__getitem__(idx)
        elif isinstance(index, int):
            index = self.wrapped_index(index)
            return super().__getitem__(index)
        else:
            print("RAISING EXCEPTION")
            raise ValueError

    def __setitem__(self, index, item):
        index = self.wrapped_index(index)
        return super().__setitem__(index, item)

    def __array_finalize__(self, obj):
        if obj is None: return
        self.grid_size = getattr(obj, 'grid_size', obj.size)
        pass

    def wrapped_index(self, index):
        idx=None
        #if hasattr(index, '__iter__'): idx=
        if type(index) is tuple:
            idx=index[0] #only support for 1D!
        elif type(index) is int:
            idx=index
        else:
            raise ValueError('Unexpected index: {}'.format(index))

我的问题

上面的代码实际上运行良好，因此从某种意义上说我的问题得以解决。但是，我对如何在__getitem__中实现处理切片有疑问。在上面的代码中，我求助于使用循环并手动循环需要的索引并填充适当的值。这发生在Python循环中，而我想让numpy处理该循环。我认为使用numpy.take这样应该是可能的（仅查看处理切片对象的__getitem__块：

def __getitem__(self, index):
     if isinstance(index, slice):
         start=index.start
         stop=index.stop
         indices=list(range(start, stop))
         return super().take(self, indices, mode='wrap')

但是，现在我遇到了一个错误：

>>> b[1:5]
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
  File "path/to/class/file", line 130, in __getitem__
    return super().take(self, indices, mode='wrap')
TypeError: 'list' object cannot be interpreted as an integer

据我所知，根据documentation中的第一个示例，允许将索引列表传递到take。我怀疑自己使用的super()错误，但不确定。感谢您的任何帮助。

在narray子类中使用super（）。take包装numpy切片

0 个答案: