TypeError：'generator'對象沒有屬性'__getitem__'

Question

我寫了一個應該返回字典的生成函數。 但是當我嘗試打印字段時，我收到以下錯誤

print row2['SearchDate']
TypeError: 'generator' object has no attribute '__getitem__'

這是我的代碼

from csv import DictReader
import pandas as pd
import numpy as np


def genSearch(SearchInfo):
    for row2 in DictReader(open(SearchInfo)):
        yield row2

train = 'minitrain.csv'

SearchInfo = 'SearchInfo.csv'

row2 = {'SearchID': -1}

for row1 in DictReader(open(train)):
    if 'SearchID' in row1 and 'SearchID' in row2 and row1['SearchID'] == row2['SearchID']:
        x = deepcopy( row1 )
        #x['SearchDate'] = row2['percent']
        x.update(row2)    
        print 'new'
        print x
    else: 
        #call your generator
        row2 = genSearch(SearchInfo)
        print row2['SearchDate']

Answer 1

Generator返回一個迭代器，你顯然需要在它上面調用next。

你的最后一行代碼應該是這樣的 -

rows_generator = genSearch(SearchInfo)
row2 = next(rows_generator, None)
print row2['SearchDate']

理想情況下，我們在循環中使用迭代器，它會自動為我們做同樣的事情。

Answer 2

生成器必須是迭代器 ，而不是迭代器 。 Iterables包含__item__()和__getitem__()方法，而迭代器包含next() / __next__()方法（python版本2.x / 3.x）。