Làm thế nào để hỏi một iterator trong python mà không thay đổi trạng thái yêu cầu trước của nó

Question

Tôi xây dựng một đối tượng có thể lặp lại A chứa danh sách các đối tượng khác B. Tôi muốn có thể tự động bỏ qua một đối tượng cụ thể B trong danh sách nếu nó là bị gắn cờ xấu khi đối tượng A được sử dụng trong vòng lặp for.

class A(): 
    def __init__(self): 
    self.Blist = [B(1), B(2), B(3)] #where B(2).is_bad() is True, while the other .is_bad() are False 

    def __iter__(self): 
    nextB = iter(self.Blist) 
    #if nextB.next().is_bad(): 
    # return after skip 
    #else: 
    # return nextB 

Tuy nhiên, tôi không thể tìm ra cách để viết các điều kiện được nhận xét trong mã giả ở trên, không được dùng cách lặp hỏi (mệnh đề vẫn thất bại)

Cảm ơn!

Answer 1

Bạn có thể sử dụng một chức năng máy phát điện:

def __iter__(self): 
     for item in self.Blist: 
      if not item.is_bad(): 
       yield item 

Một chức năng máy phát điện được đánh dấu bằng các từ khóa yield. Một hàm máy phát trả về một đối tượng máy phát, là một trình lặp. Nó sẽ đình chỉ thực hiện tại tuyên bố yield và sau đó tiếp tục xử lý khi các cuộc gọi thường gọi next trên interator.

Answer 2

Làm thế nào về:

def __iter__(self): 
    nextB = iter(self.Blist) 
    for b_obj in nextB: 
     if b_obj.is_bad(): 
      yield b_obj 

Một ví dụ đơn giản:

class B: 
    def __init__(self, cond): 
     self.cond = cond 

    def is_bad(self): 
     return self.cond 

class A: 
    def __init__(self): 
    self.Blist = [B(True), B(False), B(True)] 

    def __iter__(self): 
    nextB = iter(self.Blist) 
    for b_obj in nextB: 
     if b_obj.is_bad(): 
      yield b_obj 

a = A() 
for x in a: 
    print(x.is_bad()) 

>> True 
    True