Python生成器和文件對象

Question

我有關於生成器和文件對象的問題。

如果我寫代碼

def iter(ls):
    count=0
    for k in ls:
        if count==2:
            break
        count+=1
    for k in ls:
        yield k

test=[1,2,3,4]

k=iter(test)
for i in k:
    print i

然后輸出是

1 2 3 4

是的，我可以理解這一點，因為此代碼中的break語句不會影響以下yield語句。

但是如果我對文件執行類似的操作，在這種情況下，我會創建一個“數據”文件並將其寫入

1
2
3
4

如果我只改變線

test=[1,2,3,4]  →  test=open("data","r")

然后結果是

4

這意味着break語句以某種方式影響，因此for循環中的genetor“yield”與前一種情況不同。

造成這種差異的原因是什么？

Answer 1

是的，因為文件對象也存儲了當前讀取的位置。 因此，當您嘗試再次迭代文件對象時（不執行.seek() ），它將從下一行（您離開的位置）開始。 那是在打開文件並迭代一次之后，如果你再次嘗試迭代它（不做.seek() ），它就不會產生任何結果，因為在第一次迭代中光標位於文件的末尾而且沒有別的東西可讀。 示例 -

我的a.txt -

1
2
3
4

代碼 -

>>> for i in f:
...     print(i)
...
1

2

3

4
>>> for i in f:
...     print(i)
...
>>>

-

如上所述，如果要將文件光標移動到文件的開頭，則必須使用file.seek(0) ，0表示移動起始位置。