Класс Iterator не может поднять StopIteration для данных HDF5, открытых с помощью h5py - PullRequest
Новая
       178

Класс Iterator не может поднять StopIteration для данных HDF5, открытых с помощью h5py

0 голосов
/ 27 мая 2020

Я пытаюсь реализовать итеративный класс для набора данных h5. 

class Argh():
    def __init__(self, data):
        self.data = data
        self.c_idx = 0 

    def __getitem__(self, idx):
        return self.data[idx]

    def __len__(self):
        return len(self.data)

    def __next__(self):
        try:
            x = self.data[self.c_idx]
        except IndexError or ValueError:
            raise StopIteration
        self.c_idx += 1
        return x

    def __iter__(self):
        return self

Когда я пытаюсь go через него, как через последовательность, он не может вызвать исключение StopIteration перед ValueError или IndexError. Этого не происходит, если я просто перебираю набор данных h5 напрямую или использую итерацию путем индексации и getitem: 

with h5py.File('test.h5', 'w') as f:
     f.create_dataset(name='matrix', data=np.arange(10), dtype=np.float32)

f =  h5py.File('test.h5', 'r')
A = Argh(np.arange(0,10))
B = Argh(f['matrix'])

for x in A: pass
for x in B.data: pass
for i in range(len(B)): B[i]
for x in f['matrix']: pass
for x in B: pass

ValueError  Traceback (most recent call last)
<ipython-input-7-1dcb814e7a79> in <module>
      3 for i in range(len(B)): B[i]
      4 for x in f['matrix']: pass
----> 5 for x in B: pass
ValueError: Index (10) out of range (0-9)

Я тестировал это на нескольких других объектах, включая массивы zarr но наблюдал такое поведение только для h5py открытых наборов данных h5.

1 Ответ

0 голосов
/ 27 мая 2020

IndexError or ValueError - выражение, результат которого равен IndexError. Синтаксис для сопоставления нескольких типов исключений: 

except (IndexError, ValueError):
...