补全get_entity函数以获取指定索引的完整实体（含前后关联成分）

问题分析 #

当前get_entity函数仅能获取单个前置祖先成分，无法完整拼接包含多段前置/后续关联的实体（如索引24对应的“Christian”需完整获取为“Hans Christian Andersen”），核心问题是前置关联的遍历逻辑仅执行一次，未循环获取所有关联成分。

修复后的完整代码 #

def get_entity(triples, idx):
    entity_parts = []
    
    # 处理前置关联成分：循环获取所有前置关联节点，直到无前置为止
    current_idx = idx
    while True:
        # 查找以当前节点为object的三元组（即前置关联的subject）
        prev_relations = [triple for triple in triples if triple[2] == current_idx]
        if not prev_relations:
            break
        # 语义三元组中单个节点通常仅对应一个前置关联，取第一个结果
        current_idx = prev_relations[0][0]
        # 将前置成分插入到列表开头，保证实体顺序正确
        entity_parts.insert(0, triples[current_idx][1])
    
    # 添加当前索引对应的核心成分
    entity_parts.append(triples[idx][1])
    
    # 处理后续关联成分：循环获取所有后续关联节点，直到无后续为止
    current_idx = idx
    while True:
        # 查找以当前节点为subject的三元组（即后续关联的object）
        next_relations = [triple for triple in triples if triple[0] == current_idx]
        if not next_relations:
            break
        current_idx = next_relations[0][2]
        # 将后续成分追加到列表末尾
        entity_parts.append(triples[current_idx][1])
    
    # 拼接所有成分得到完整实体
    return ' '.join(entity_parts)

关键修改说明 #

• 前置关联遍历：将原单次查找逻辑改为循环，持续追溯当前节点的所有前置关联节点，直到没有前置节点为止，每次将前置成分插入结果列表头部，确保实体的语序正确。
• 后续关联遍历：同样采用循环遍历所有后续关联节点，将后续成分追加到结果列表尾部，保证完整拼接所有关联的后续成分。

示例验证 #

以索引24对应“Christian”为例：

1. 循环查找前置关联，找到对应“Hans”的节点，将“Hans”插入列表头部；
2. 添加核心成分“Christian”；
3. 循环查找后续关联，找到对应“Andersen”的节点，将“Andersen”追加到列表尾部；
4. 最终拼接得到完整实体：Hans Christian Andersen。

内容的提问来源于stack exchange，提问作者panda