python的迭代操作和in运算符重载问题

streethacker

最近在学习<Data Structures and Algorithms Using Python>一书，在数组这一章，定义二维数组的时候遇到一个问题，二维数组定义如下：

--------------------------------------------------------------------二维数组-----------------------------------------------------------------------------------------
from array import Array

class Array2D:
        def __init__(self, numRows, numCols):
                self._theRows = Array(numRows)
               
                for i in range(numRows):
                                self._theRows = Array(numCols)

        def numRows(self):
                return len(self._theRows)

        def numCols(self):
                return len(self._theRows[0])

        def clear(self, value):
                for row in self._theRows:
                                row.clear(value)

        def __getitem__(self, ndxTuple):
                assert len(ndxTuple) == 2, "Invalid number of array subscripts."
                row = ndxTuple[0]
                col = ndxTuple[1]
                assert row >= 0 and row < self.numRows()\
                        and col >= 0 and col < self.numCols, \
                        "Array subscript out of range."
                return self._theRows[row][col]

        def __setitem__(self, ndxTuple, value):
                assert len(ndxTuple) == 2, "Invalid number of array subscripts."
                row = ndxTuple[0]
                col = ndxTuple[1]
                assert row >= 0 and row < self.numRows() \
                        and col >= 0 and col < self.numCols(),\
                        "Array subscript out of range."
                self._theRows[row][col] = value

if __name__ == "__main__":
                arr2d = Array2D(3,4)
                arr2d.clear(100)
                print arr2d[1,2]
               
--------------------------------------------------------------------二维数组-----------------------------------------------------------------------------------------

测试代码的时候，总是报错：
Traceback (most recent call last):
  File "array2d.py", line 40, in <module>
    arr2d.clear(100)
  File "array2d.py", line 17, in clear
    for row in self._theRows:
TypeError: instance has no next() method


self._theRows是一维数组的引用，一维数组定义代码如下：

--------------------------------------------------------------------一维数组-----------------------------------------------------------------------------------------
import ctypes

class _ArrayIterator:
        def __init__(self, theArray):
                self._arrayRef = theArray
                self._curNdx = 0
       
        def __iter__(self):
                return self

        def __next__(self):
                if self._curNdx < len(self._arrayRef):
                                item = self._arrayRef[self._curNdx]
                                self._curNdx += 1
                                return item
                else:
                                raise StopIteration


class Array:
        def __init__(self, size):
                assert size > 0, "Array size must be > 0"
                self._size = size
                PyArrayType = ctypes.py_object * size
                self._elements = PyArrayType()
                self.clear(None)

        def __len__(self):
                return self._size

        def __getitem__(self, index):
                """
                When the subscript notation is used in a program,
                y = x, Python will call the __getitem__ method,
                passing the value of i to the index parameter.
                """
                assert index >= 0 and index < self._size, "Array subscript out of range"
                return self._elements[index]

        def __setitem__(self, index, value):
                assert index >= 0 and index < self._size, "Array subscript out of range"
                self._elements[index] = value

        def clear(self, value):
                for i in range(self._size):
                                self._elements = value

        def __iter__(self):
                return _ArrayIterator(self._elements)


--------------------------------------------------------------------一维数组-----------------------------------------------------------------------------------------

一维数组中我已经定义了迭代，而self._theRows又是一维数组的引用，为什么还是会报错：没有next方法呢？这里in运算符应该还需要一个__contains__方法，我尝试添加过，但还是同样的报错。另外__contains__方法也不是非常理解。请各位帮忙指点，不胜感激。

a-plus

楼主一定是用的Python2.x运行的吧。而且traceback不是明显告诉你没有next方法吗？  __next__()方法是在Python3.x里用的，next()才是在Python2.x里用的。

streethacker

回复 2# a-plus
果真是这个问题，之前学习python就是用的python 2.7, 学习这本书的时候因为书中使用的是python 3，在两个之间切换来切换去的把自己都绕晕了。非常感谢。

关于__contains__方法我还想追问一下：为什么不需要在上面的类中定义__contains__方法，就可以直接使用in运算符呢？对__contains__方法和in运算符的重载始终没怎么理解。

   

laike9m

直接用3+就行了