如何在循环中使用非类型函数调用生成器类型函数？

Question

我在 Reader() 类中有一个生成器函数 getElements，它从 xml 文件中生成所有元素。我还想要一个函数 getFeatures ，它只生成带有功能标签的元素。

我的尝试是在调用 getFeatures 时将标志 featuresOnly 设置为 True，并在 getFeatures 中调用 self.getElements，如下所示：

def getFeatures(self):
    self.getFeaturesOnly = True
    self.getElements()

这样在 getElements() 中我只需要做

def getElements(self):
    inFile = open(self.path)
    for element in cElementTree.iterparse(inFile):
        if self.getFeaturesOnly == True:
            if element.tag == 'feature':
                yield element
        else:
            yield element
     inFile.close()

但是，当我这样做并运行时我

 features = parseFeatureXML.Reader(filePath)
 for element in features.getFeatures():#
       print element

得到： TypeError: 'NoneType' object is not iterable 这是因为 getFeatures 不包含收益。现在，我知道如何解决这个问题的方法是将 getElements 的代码复制到 getFeatures 中，并仅

if elementFunctions.getElmentTag(element) == 'feature':

在 getFeatures() 函数中使用，但我不想重复任何代码。那么我怎样才能保持生成器功能，并拥有一个不同的功能，我只指定我想要获得的标签？

Answer 1

首先要做的事情是：你有这个错误，因为你没有返回生成器

这意味着你必须更改：

def getFeatures(self):
    self.getFeaturesOnly = True
    self.getElements()

with：

def getFeatures(self):
    self.getFeaturesOnly = True
    return self.getElements()    # returning the generator

清除了这个，TBH我不会设计我的Reader()类就像这样。

我会让 getElement 产生所有元素：

def getElements(self):
    inFile = open(self.path)
    for element in cElementTree.iterparse(inFile):
        yield element
    inFile.close()

然后 getFeatures() 进行过滤：

def getFeatures(self):
    for element in self.getElements():
        if element.tag == 'feature':
            yield element

Answer 2

您收到 TypeError 的原因并不是 getFeatures 不包含收益，而是因为 getFeatures 不返回任何内容< /em>.如果您希望 getFeatures 返回由 getElements 返回的迭代器，则必须使用 return：

def getFeatures(self):
    self.getFeaturesOnly = True
    return self.getElements()

当您使用它时，您确实应该' t do if expr == True;只需执行 if expr 即可，即使 expr 为 true（概念）但不是 True （对象）。将仅功能支持提升到 getElements 中，更常见的方法是在 getFeatures 本身中执行此操作，如下所示：

def getFeatures(self):
    for element in self.getElements():
        if element.tag == 'feature':
            yield element

def getElements(self):
    inFile = open(self.path)
    for element in cElementTree.iterparse(inFile):
        yield element
    inFile.close()