파이썬은

Question

from collections.abc import Sequence 

class Map(Sequence): 
    """ Represents a map for a floor as a matrix """ 

    def __init__(self, matrix): 
     """ Takes a map as a matrix """ 
     self.matrix = matrix 
     self.height = len(matrix) 
     self.width = len(matrix[0]) 
     super().__init__() 

    def __getitem__(self, item): 
     """ Needed by Sequence """ 
     return self.matrix[item] 

    def __len__(self): 
     """ Needed by Sequence """ 
     return len(self.matrix) 

    def search(self, entity): 
     """ Returns a generator of tuples that contain the x and y for every element in the map that matches 'entity' """ 
     for row in range(self.height): 
      for column in range(self.width): 
       if matrix[row][column] == entity: 
        yield (row, column) 


# Examples 

gmap = Map([[0, 0, 0], 
      [0, 1, 0], 
      [0, 0, 0]]) 

for entity in gmap: 
    print(entity) 

이 어떻게 그렇게 그

for entity in gmap: 
    print(entity) 

수율 0 0 0 0 1 0 0 0 0하지

[0, 0, 0] 
[0, 1, 0] 
[0, 0, 0] 

이 Sequence를 서브 클래 싱 할 필요에서 저를 절약 할 수 __iter__을 구현할 수 있습니다 매트릭스 클래스 반복하고 코드를 만들 것 search() neater

또한, 내가 사용해야 할 다른 마법 방법이 있습니까? 당신과 같이 __iter__()을 구현할 수

Answer 1

(나는 작업을 반복 취득 후 옆 __str__에서, 메신저 그 일) :

from itertools import chain 

def __iter__(self): 
    return chain.from_iterable(self.matrix) 

itertools.chain.from_iterable()은 반복 가능 객체의 반복자를 받아 모두 함께 결합되어 있습니다. 생성기를 생성하므로 추가 메모리를 사용하지 않습니다.