对内部 numpy 数组进行类迭代？任务？

Question

如何让类通过其内部 numpy 数组进行迭代： 只是想法不起作用：

class ABC:

  def __init__(self):
     self.ary = np.zeros(50)

  def __iter__(self): return np.nditer(self.ary)
  def next(self): ...??..

还有如何让分配工作：

abc = ABC()
abc[5] = 12
abc[7:9] 0

Answer 1

来自文档，

iterator.__next__()：

从容器中返回下一个项目。如果没有其他项目，则引发 StopIteration 异常。此方法对应 Python/C API 中 Python 对象的类型结构的 tp_iternext 槽。

对于容器类的setting 和getting 值，您需要实现__getitem__ 和__setitem__。

为您的示例代码

class ABC():
  def __init__(self):
     self.ary = np.zeros(50)
     self.index = self.ary.shape[0]

  def __iter__(self): 
     return np.nditer(self.ary)

  def next(self):
      if self.index == 0:
          raise StopIteration
      self.index = self.index - 1
      return self.data[self.index]

  def _check_indx(self, idx):
      if abs(idx) >= self.ary.shape[0]:
          raise IndexError(f"Invalid Index {idx} for array with shape {self.ary.shape}")

  def __setitem__(self, idx, value):
      self._check_indx(idx)
      self.ary[idx] = value

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