Обертывание кусочков с помощью super (). Взять в подкласс Narray - PullRequest
Новая
       21

Обертывание кусочков с помощью super (). Взять в подкласс Narray

1 голос
/ 23 апреля 2019

Я пытаюсь создать подкласс пустого массива, чтобы создать массивы с периодическими граничными условиями.Мне это нужно только для работы с 1D массивами.

Что я хочу : Предположим, у меня есть объект 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

Насколько я могу судить, передача списка индексов в take разрешена, судя по первому примеру в документации . У меня есть подозрения, что я использую super() неправильноно я не уверен. Любая помощь здесь приветствуется.

...