Lấy mục thứ n của máy phát điện bằng Python

Question

Có cách viết ngắn gọn hơn về cú pháp sau đây không?

gen = (i for i in xrange(10)) 
index = 5 
for i, v in enumerate(gen): 
    if i is index: 
     return v 

Có vẻ như gần như tự nhiên rằng trình tạo nên có biểu thức gen[index], hoạt động như một danh sách, nhưng có chức năng giống với mã ở trên.

Answer 1

một phương pháp sẽ được sử dụng itertools.islice

>>> next(itertools.islice(xrange(10), 5, 5 + 1)) 
5 

Answer 2

Có lẽ bạn nên giải thích thêm về trường hợp sử dụng thực tế.

>>> gen = xrange(10) 
>>> ind=5 
>>> gen[ind] 
5 

Answer 3

Bạn có thể làm điều này, sử dụng count làm ví dụ máy phát điện:

from itertools import islice, count 
next(islice(count(), n, n+1)) 

Answer 4

bạn chỉ có thể chuyển đổi các máy phát điện vào một danh sách và sử dụng chỉ mục như bình thường:

>>> [i for i in range(10)][index] 
5 

Answer 5

Tôi muốn chống lại sự cám dỗ để xử lý các máy phát điện như danh sách. Cách tiếp cận đơn giản nhưng ngây thơ là một lớp lót đơn giản:

gen = (i for i in range(10)) 
list(gen)[3] 

Nhưng hãy nhớ rằng, máy phát không giống như danh sách. Họ không lưu trữ kết quả trung gian của họ ở bất cứ nơi nào, vì vậy bạn không thể đi ngược lại. Tôi sẽ chứng minh vấn đề với một ví dụ đơn giản trong repl python:

>>> gen = (i for i in range(10)) 
>>> list(gen)[3] 
3 
>>> list(gen)[3] 
Traceback (most recent call last): 
    File "<stdin>", line 1, in <module> 
IndexError: list index out of range 

Khi bạn bắt đầu trải qua một máy phát điện để có được giá trị thứ n trong dãy, máy phát điện tại là ở tiểu bang khác, và cố gắng nhận lại giá trị thứ n sẽ trả lại cho bạn một kết quả khác, điều này có khả năng dẫn đến lỗi trong mã của bạn.

Hãy xem một ví dụ khác, dựa trên mã từ câu hỏi.

Ban đầu, người ta sẽ mong đợi những điều sau đây để in 4 hai lần.

gen = (i for i in range(10)) 
index = 4 
for i, v in enumerate(gen): 
    if i == index: 
     answer = v 
     break 
print(answer) 
for i, v in enumerate(gen): 
    if i == index: 
     answer = v 
     break 
print(answer) 

nhưng gõ này vào repl và bạn nhận được:

>>> gen = (i for i in range(10)) 
>>> index = 4 
>>> for i, v in enumerate(gen): 
...  if i == index: 
...    answer = v 
...    break 
... 
>>> print(answer) 
4 
>>> for i, v in enumerate(gen): 
...  if i == index: 
...    answer = v 
...    break 
... 
>>> print(answer) 
9 

Chúc may mắn truy tìm lỗi xuống.

Answer 6

Điều đầu tiên mà đến tâm trí của tôi là:

gen = (i for i in xrange(10)) 
index = 5 

for i, v in zip(range(index), gen): pass 

return v