写访问迭代器？或者：将值发送到numpy数组的生成器

Question

我是Python的迭代器的新手，所以也许我的语言并不总是正确的。

我有一个类包装的numpy.ndarray列表：

class wrapper: 
    def __init__(self, myList): 
     self.myList = myList 

    def getArrayIterator(self): 
     for arr in self.myList: 
      yield arr 

#set list of arrays in wrapper 
myList = [rand(3,3), rand(3,3), rand(3,3)] 
w = wrapper(myList) 

正如我的理解是，该第二方法返回发电机。 现在我想使用发电机来遍历列表和阵列恢复到别的东西：

for a in w.getArrayIterator(): 
    a = zeros((3,4)) 

我希望有通过这里引用语义通，但似乎并不如此。

所以我试图用Python的send()我getArrayIterator功能：

# ... 
def getArrayIterator(self): 
    for arr in self.myList: 
     val = (yield arr) 
     if val is not None: 
      arr = val 
# ... 

但是，这不会工作，要么是因为：

a.send(zeros((3,4))) 
    AttributeError: 'numpy.ndarray' object has no attribute 'send' 

有一个简单的解决方案，实现我的期望行为？ 我错过了什么吗？

---

编辑：有人向我指出，我应该提供有关我的实际问题的更多信息。上面的例子当然简化了。

我有一个列表numpy.ndarray列表表示一个张量T封装在我的班级。当访问T的一个元素时：t_ijkl我需要将存储在列表中的矩阵相乘：A(i)*B(j)*C(k)*D(l)，第一个和最后一个是行/列向量。

所以有一组A，一组B，等等。每个属于我的应用程序中的一个网格点。

我现在想要在与每个网格点关联的矩阵上具有遍历所有网格点和迭代器的迭代器。

想到的第一个想法是使用迭代器C++风格来读取和写入矩阵。但正如agf指出的那样，这不是一个真正可行的方法。所以我想我会使用不同的迭代器来读取访问权限和专门的setter方法来为矩阵设置新的值。

Answer 1

我假设这是一个简化的例子，因为你有类的地方似乎没有理由不使用普通列表。

当你

name = somelist[0] 
name = 'other' 

你的对象 somelist指向名称 name第一个指数，然后在对象 'other'指向名称 name

。你永远不会指向somelist[0]在'other'。

所以除了send，您需要实际分配到列表：

class wrapper: 
    def __init__(self, myList): 
     self.myList = myList 

    def getArrayIterator(self): 
     for i, arr in enumerate(self.myList): 
      yarr = (yield arr) 
      if yarr is not None: 
       self.myList[i] = yarr 

#set list of arrays in wrapper 
myList = [1, 2, 3, 4, 5] 
w = wrapper(myList) 
witer = w.getArrayIterator() 
try: 
    a = next(witer) 
    for i in range(6, 11): 
     print a, 
     a = witer.send(i) 
except StopIteration: 
    pass 
print 
print w.myList 

Answer 2

您只能使用发电机（yield关键字）读值不会写

Answer 3

听起来也许你想实施__getitem__和__setitem__。

class wrapper: 
    def __init__(self, myList): 
     self.myList = myList  

    def __iter__(self): 
     for arr in self.myList: 
      yield arr  

    def __getitem__(self, key): 
     return self.myList[key] 

    def __setitem__(self, key, value): 
     self.myList[key] = value 

用法：

for index, arr in enumerate(w): 
    same_arr = w[index] 
    w[index] = zeros((3,4)) 

有很多其他有用的功能，如果你认为它会让你的生活更轻松，你可以实现。以下是关于模拟容器类型的documentation。