無法在同一個循環中使用來自兩個迭代器的值

Question

我有以下python腳本，我想從兩個迭代器中選擇值。

filename = "small" 
with open(filename,'r') as plot_data: 
    main_dict = dict() 
    line_one = itertools.islice(plot_data, 0, None, 4) 
    line_two = itertools.islice(plot_data, 2, None, 4) 
    dictionary = defaultdict(list) 
    #take values from iterators alternatively. 
    for movie_name, movie_plot in itertools.izip(line_one, line_two): 
     movie_plot = movie_plot.lower() 
     words = re.findall(r'\w+', movie_plot, flags = re.UNICODE | re.LOCALE) 
     elemStopW = filter(lambda x: x not in stopwords.words('english'), words) 
     #list of words. 
     print elemStopW 
     for word in elemStopW: 
      word = PorterStemmer().stem_word(word) 
      dictionary[movie_name].append(word) 
      main_dict[word] = len(main_dict) 
    print main_dict 

此腳本不打印任何內容。我不懂爲什麼。我不想合併迭代器，因爲我想在同一個循環中使用兩個值。

任何幫助表示讚賞。

編輯：爲了避免一些通關（如在評論中）。下面的腳本工作正常

filename = "small" 
with open(filename,'r') as plot_data:      
     main_dict = dict() 
     line_one = itertools.islice(plot_data, 0, None, 4) 
     dictionary = defaultdict(list)                  
     for movie_name in line_one:               
       print movie_name 

Answer 1

這可能不是你所期望的：

line_one = itertools.islice(plot_data, 0, None, 4) 
line_two = itertools.islice(plot_data, 2, None, 4) 

由於plot_data是一個文件對象，無論從迭代器讀取將推動文件 - 它仍然會順序讀取而不是平行讀兩遍。

您可以使用itertools.tee複製文件迭代器，以便它可以並行讀取兩次：

plot1, plot2 = itertools.tee(plot_data, 2) 
line_one = itertools.islice(plot1, 0, None, 4) 
line_two = itertools.islice(plot2, 2, None, 4) 

注意，如果兩個迭代器的位置可以彼此遠離漂移，這可能需要一個大量的內存，你最好打開文件兩次。在這種情況下，這應該不成問題。