在python中包装for循环

Question

我想包装 for 循环的功能，这样使用起来会更直观。就我而言，我必须解析 xml 文件，如下所示：

<instance id="line-n.w8_047:15696:">
    <answer instance="line-n.w8_047:15696:" senseid="product" />
    <context> context1 </context>
</instance>

<instance id="line-n.w8_088:12441:">
    <answer instance="line-n.w8_088:12441:" senseid="product" />
    <context> another context</context>
</instance>

我为Instance 写了一个这样的类：

class Instance:
    def __init__(self, id, answer, context):
        self.id = id
        self.answer = answer
        self.context = context

我编写了下一个函数来枚举实例：

import xml.etree.ElementTree as ET
def enum_instances(file_path, action):
    for instance_xml in ET.parse(file_path).getroot().find('lexelt'):
        action(Instance(
            instance_xml.attrib['id'],
            instance_xml.find('answer').attrib['senseid'],
            instance_xml.find('context').text)
        )

action 参数是对Instance 执行某些操作的回调，如下所示：

enum_instances('/path/to/xml', lambda instance: print(instance.context))

但它看起来有点奇怪，我希望它更直观，像这样：

for instance in enum_instances(file_path):
    print(instance.context)

实现“可迭代”功能的最佳方式是什么？ 谢谢

Answer 1

在这种情况下，获胜的发电机。类似的东西

def enum_instances(file_path):
    for instance_xml in ET.parse(file_path).getroot().find('lexelt'):
        yield Instance(
            instance_xml.attrib['id'],
            instance_xml.find('answer').attrib['senseid'],
            instance_xml.find('context').text)

那么你就可以准确地说出你的要求了：

for instance in enum_instances(file_path):
    print(instance.context)

Answer 2

不要将可调用对象传递给你的函数，而是让它产生一个Instance，然后你可以获得你想要的行为，例如：

def enum_instances(file_path, action=None):
    for instance_xml in ET.parse(file_path).getroot().find('lexelt'):
        instance = Instance(
            instance_xml.attrib['id'],
            instance_xml.find('answer').attrib['senseid'],
            instance_xml.find('context').text)
        if action is not None:
            instance = action(instance)
        yield instance

在这种情况下 - 我已将 action 默认为 None，但如果需要，您可以在 yield 之前传递一个函数以某种方式改变 instance。

然后：

for instance in enum_instances('some_file_path'):
    print(instance.context)