将CSV中N行数据转XML遇索引错误，求适配任意行数的解决方案

错误原因分析 #

• 索引越界错误：convert_row函数中使用了row[2]，但你的CSV每行只有2列（ColA、ColB），不存在索引2的元素，直接触发IndexError。
• XML结构错误：原代码试图为每行生成<Class>标签，而目标格式是单个<Class>根节点包含所有<Id>子节点。
• 未处理表头：原代码没有跳过CSV的第一行表头，会把表头内容也写入XML。

修正后的完整代码 #

import csv

def convert_to_xml(csv_file_path, xml_file_path):
    # 读取CSV数据，跳过表头
    with open(csv_file_path, 'r') as csv_f:
        csv_reader = csv.reader(csv_f)
        # 跳过第一行表头
        next(csv_reader)
        # 收集所有Id值，过滤空行避免异常
        ids = [row[1] for row in csv_reader if row]

    # 生成符合要求的XML内容
    xml_content = ['<Class>']
    for id_val in ids:
        xml_content.append(f'    <Id>{id_val}</Id>')
    xml_content.append('</Class>')

    # 写入XML文件
    with open(xml_file_path, 'w') as xml_f:
        xml_f.write('\n'.join(xml_content))

# 调用函数，替换为你的实际文件路径
convert_to_xml('Class.csv', 'output.xml')

代码关键点说明 #

• 安全文件操作：用with语句自动管理文件生命周期，避免手动关闭文件导致的资源泄漏。
• 适配任意行数：通过遍历CSV所有行收集Id值，无论CSV有多少行都能处理，同时过滤空行防止报错。
• 正确XML结构：先构建根节点<Class>，再逐个添加<Id>子节点，最后闭合根节点，完全匹配目标格式。
• 跳过表头：使用next(csv_reader)跳过CSV第一行的表头数据，避免无效内容写入XML。

测试结果 #

针对你提供的示例CSV，运行代码后会生成如下XML：

<Class>
    <Id>1001</Id>
    <Id>1021</Id>
    <Id>1024</Id>
</Class>

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