如何用Python定位同名XML元素并修改指定属性值

Got it, let's walk through how to tackle this. We'll use Python's built-in xml.etree.ElementTree module first since it's part of the standard library—no extra installs needed. Then I'll cover some useful extensions you might want.

Basic Solution #

First, let's take your XML content (fixed the minor typo in the root element) and write code to locate the <fruit_type> element with fruit_type="0x" and update its attribute value.

Code Example #

import xml.etree.ElementTree as ET

# Option 1: Parse from a file
# tree = ET.parse('your_file.xml')
# root = tree.getroot()

# Option 2: Parse from a string (using your provided XML)
xml_content = '''<?xml version="1.0"?>
<apple view_filter="simple" version="1" format="1">
  <apples fruit_id="3" type="red" name="american">
    <basket version="1" type="6" pieces="12" expiration="12">
      <fruit_type colour="000" fruit_type="0x" weight="32"></fruit_type>
    </basket>
  </apples>
</apple>'''
root = ET.fromstring(xml_content)

# Locate the target element using XPath
target_element = root.find(".//fruit_type[@fruit_type='0x']")

if target_element is not None:
    # Modify the attribute value (e.g., change to "0y")
    target_element.set('fruit_type', '0y')
    print("Attribute updated successfully!")
    
    # Write the modified XML back to a file (if using file parse)
    # tree.write('modified_file.xml', encoding='utf-8', xml_declaration=True)
    
    # If using string parse, convert back to string to see the result
    modified_xml = ET.tostring(root, encoding='utf-8').decode('utf-8')
    print(modified_xml)
else:
    print("No element found with fruit_type='0x'")

Explanation #

• Parsing: We either load from a file with ET.parse() or from a string with ET.fromstring().
• XPath Query: .//fruit_type[@fruit_type='0x'] looks for any <fruit_type> element anywhere in the XML where the fruit_type attribute equals "0x". The .// means search recursively through all child elements.
• Modifying Attribute: target_element.set('fruit_type', 'new_value') updates the attribute directly.
• Saving Changes: If you parsed from a file, use tree.write() to save the modified XML back to disk. For string input, ET.tostring() converts the tree back to a human-readable XML string.

Useful Extensions #

Here are some common extensions you might need for more robust functionality:

• Update All Matching Elements (Not Just the First)
  If there are multiple <fruit_type> elements with fruit_type="0x", use findall() instead of find() to get all matches and update each one:

  target_elements = root.findall(".//fruit_type[@fruit_type='0x']")
  for elem in target_elements:
      elem.set('fruit_type', '0y')
      print(f"Updated element: {elem}")
  

• Validate New Attribute Values
  Add checks to ensure the new value meets your requirements (e.g., allowed format or values):

  new_value = "0y"
  allowed_values = ["0y", "0z", "1a"]
  
  if new_value in allowed_values:
      target_element.set('fruit_type', new_value)
  else:
      print(f"Invalid value: {new_value}. Allowed values are {allowed_values}")
  

• Using lxml for Advanced XPath and Pretty Printing
  If you need more powerful XPath features (like namespace handling) or want nicely formatted output, use the lxml library (install with pip install lxml):

  from lxml import etree
  
  # Parse XML
  tree = etree.fromstring(xml_content.encode('utf-8'))
  
  # Find target element
  target_element = tree.xpath(".//fruit_type[@fruit_type='0x']")[0]
  
  # Modify attribute
  target_element.set('fruit_type', '0y')
  
  # Pretty print the modified XML
  print(etree.tostring(tree, pretty_print=True, encoding='utf-8').decode('utf-8'))
  

• Error Handling for File Operations
  Add try-except blocks to handle common issues like missing files or invalid XML:

  try:
      tree = ET.parse('your_file.xml')
      root = tree.getroot()
      
      # Modify attribute here...
      
      tree.write('modified_file.xml', encoding='utf-8', xml_declaration=True)
      print("File saved successfully!")
  except FileNotFoundError:
      print("Error: Input file not found.")
  except ET.ParseError:
      print("Error: Invalid XML format.")
  except Exception as e:
      print(f"An unexpected error occurred: {str(e)}")
  

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