使用 python ET 跟踪 XML 跟踪文件并查找索引。痕迹

Question

我有一个 XML 跟踪文件，它有不同的标签。我正在尝试在文件中查找 X 和 Y 值来计算运动矢量。我将部分文件放在下面：

<Picture id="1" poc="1">
    <GOPNr>0</GOPNr>
    <SubPicture structure="0">
        <Slice num="0">
            <Type>0</Type>
            <TypeString>SLICE_TYPE_P</TypeString>
            <NAL>
                <Num>6</Num>
                <Type>1</Type>
                <TypeString>NALU_TYPE_SLICE</TypeString>
                <Length>178927</Length>
            </NAL>
            <MacroBlock num="0">
                <Position>
                    <X>0</X>
                    <Y>0</Y>
                </Position>
                <QP_Y>23</QP_Y>
                <Type>5</Type>
                <TypeString>I_8x8</TypeString>
                <PredModeString>BLOCK_TYPE_I</PredModeString>
                <SkipFlag>0</SkipFlag>
            </MacroBlock>

这个文件有很多MacroBlock 标签，但其中一些标签有MotionVector。为了提取 X 和 Y 值，我使用了以下代码：

abs_x_tag=list(qpy_node.text for qpy_node in root.findall('Picture/SubPicture/Slice/MacroBlock/SubMacroBlock/MotionVector/Absolute/X'))
       

我需要找到与特定TypeString 相关的X 和Y 值，但有一次我只能使用上面的代码提取一个值。例如，我需要找到与<TypeString>P 相关的 X 和 Y 值，但我不知道该怎么做。你能帮我解决这个问题吗？ 谢谢

Answer 1

当您想要解析一个块中的多个值时，特别是如果您必须根据其中一个或多个值做出决定，请逐块解析。如果您分别提取值，则会失去它们之间的链接

data="""<Picture><MacroBlock num="936">
                    <MotionVector list="0">
                        <Absolute>
                            <X>0</X>
                            <Y>0</Y>
                        </Absolute>
                    </MotionVector>
                    <TypeString>P_L0_16x16</TypeString>
                </MacroBlock>
                <MacroBlock num="937">
                    <MotionVector list="0">
                        <Absolute>
                            <X>1</X>
                            <Y>2</Y>
                        </Absolute>
                    </MotionVector>
                    <TypeString>B_L0_16x16</TypeString>
                </MacroBlock></Picture>"""
                
import xml.etree.ElementTree as ET
root=ET.fromstring(data)
blocks=root.findall('MacroBlock')
for block in blocks:
    x=block.find('MotionVector/Absolute/X').text
    y=block.find('MotionVector/Absolute/Y').text
    type_string=block.find('TypeString').text[0]
    print(x, y, type_string)
>>> 0 0 P
>>> 1 2 B