Index và Slice một Generator bằng Python

Question

phép nói rằng tôi có một chức năng máy phát điện mà trông như thế này:

def fib(): 
    x,y = 1,1 
    while True: 
     x, y = y, x+y 
     yield x 

Lý tưởng nhất, tôi chỉ có thể sử dụng fib() [10] hoặc fib() [2 : 12: 2] để lấy chỉ mục và lát, nhưng hiện tại tôi phải sử dụng itertools cho những thứ này. Tôi không thể sử dụng máy phát điện để thay thế danh sách thả xuống.

tôi tin rằng các giải pháp sẽ được quấn fib() trong một lớp học:

class Indexable(object): 
    .... 

fib_seq = Indexable(fib()) 

gì nên Indexable nhìn muốn thực hiện công việc này?

Answer 1

import itertools 

class Indexable(object): 
    def __init__(self,it): 
     self.it = iter(it) 
    def __iter__(self): 
     return self.it 
    def __getitem__(self,index): 
     try: 
      return next(itertools.islice(self.it,index,index+1)) 
     except TypeError: 
      return list(itertools.islice(self.it,index.start,index.stop,index.step)) 

Bạn có thể sử dụng nó như thế này:

it = Indexable(fib()) 
print(it[10]) 
#144 
print(it[2:12:2]) 
#[610, 1597, 4181, 10946, 28657] 

ý rằng it[2:12:2] không trả lại [3, 8, 21, 55, 144] kể từ khi iterator đã tiến 11 yếu tố vì cuộc gọi đến it[10].

Edit: Nếu bạn muốn it[2:12:2] trở [3, 8, 21, 55, 144] thì có lẽ sử dụng này để thay thế:

class Indexable(object): 

    def __init__(self, it): 
     self.it = iter(it) 
     self.already_computed = [] 

    def __iter__(self): 
     for elt in self.it: 
      self.already_computed.append(elt) 
      yield elt 

    def __getitem__(self, index): 
     try: 
      max_idx = index.stop 
     except AttributeError: 
      max_idx = index 
     n = max_idx - len(self.already_computed) + 1 
     if n > 0: 
      self.already_computed.extend(itertools.islice(self.it, n)) 
     return self.already_computed[index] 

phiên bản này giúp tiết kiệm các kết quả trong self.already_computed và sử dụng những kết quả nếu có thể. Nếu không, nó tính toán nhiều kết quả hơn cho đến khi nó có đủ số để trả về phần tử hoặc lát được lập chỉ mục.

Answer 2

Vì vậy, dựa trên mã từ ~ unutbu và thêm vào một chút itertools.tee:

import itertools 

class Indexable(object): 
    def __init__(self, it): 
     self.it = it 

    def __iter__(self): 
     self.it, cpy = itertools.tee(self.it) 
     return cpy 

    def __getitem__(self, index): 
     self.it, cpy = itertools.tee(self.it) 
     if type(index) is slice: 
      return list(itertools.islice(cpy, index.start, index.stop, index.step)) 
     else: 
      return next(itertools.islice(cpy, index, index+1)) 

Answer 3

Nếu đó là một lát 1 sử dụng hơn bạn chỉ có thể sử dụng phương pháp được viết bởi ~ unutbu. Nếu bạn cần cắt nhiều lần, bạn sẽ phải lưu trữ tất cả các giá trị trung gian để bạn có thể "tua lại" trình lặp. Vì các trình lặp có thể lặp lại bất cứ thứ gì, nó sẽ không có phương thức tua lại theo mặc định.

Ngoài ra, vì một iterator tua có để lưu trữ tất cả các kết quả trung gian nó sẽ (trong hầu hết các trường hợp) không có lợi ích hơn chỉ đơn giản là làm list(iterator)

Về cơ bản ... bạn hoặc là không cần một iterator, hoặc bạn' không đủ cụ thể về tình hình.

Answer 4

Đây là câu trả lời của ~ unutbu được sửa đổi thành danh sách phân lớp. Rõ ràng là lạm dụng như append, insert v.v. sẽ tạo ra kết quả lạ!

bạn nhận được __str__ và __repr__ phương pháp miễn phí mặc dù

import itertools 
class Indexable(list): 
    def __init__(self,it): 
     self.it=it 
    def __iter__(self): 
     for elt in self.it: 
      yield elt 
    def __getitem__(self,index): 
     try: 
      max_idx=index.stop 
     except AttributeError: 
      max_idx=index 
     while max_idx>=len(self): 
      self.append(next(self.it)) 
     return list.__getitem__(self,index)