有没有办法添加/修改使用yield创建的生成器的属性？

Question

所以我想制作一个 arff reader （类似于csv 文件格式）。

我想使用 yield 来创建一个迭代器，同时也向该迭代器添加属性。

例如：

data = arff.reader(my_fname)
print data.relation
for row in data:
    print row

但在读者定义中：

def reader(fname):
    reader.relation = fname # this is assigned to the function, not the generator
    yield 1
    yield 2

有没有办法使用yield 来做到这一点，或者我是否坚持使用迭代器api？

Answer 1

你可以把它变成一个类。

class Reader(object): # Assuming Python <= 2.7
    def __init__(self, fname):
        self.fname = fname

    def __iter__(self):
        yield 1
        yield 2

r = Reader("some file")
print r.fname ## 'some file'
for line in r:
    print line ## 1 then 2