Question

该程序基本上将xml文件展平并将其写入csv。

我的问题是'values_loop'中的'row'变量没有在调用之间重置。每当我调用它时，新值都附加到旧值上。

基本上我得到了这个：

A  B  C
A  B  C  D  E  F
A  B  C  D  E  F  G  H  I

什么时候能得到这个

A  B  C
D  E  F
G  H  I

我的代码：

import csv
import xml.etree.ElementTree as ET

file_name = '2.xml'
root = ET.ElementTree(file=file_name).getroot() 
csv_file_name = '.'.join(file_name.split('.')[:-1]) + ".txt"

print csv_file_name

with open(csv_file_name, 'w') as file_:
    writer = csv.writer(file_, delimiter="\t")

    def header_loop(root, i=0, row=[]):
        for child in root:
            #print "\t"*i, child.tag.replace("{http://www.tes.com/aps/response}",""), child.attrib, i
            row.extend([child.tag.replace("{http://www.tes.com/aps/response}","")])
            header_loop(child,i+1)
        if i==0: return row

    def values_loop(root, i=0, row=[]):
        for child in root:
            #print "\t"*i, child.tag.replace("{http://www.tes.com/aps/response}",""), child.attrib, i
            row.extend([child.text])
            #print child.text
            values_loop(child,i+1)
        if i==0: return row


    #write the header 
    writer.writerow(header_loop(root[3]))

    #write the values
    writer.writerow(values_loop(root[3]))
    writer.writerow(values_loop(root[4]))

Answer 1

将“无”设置为行的默认值。

def values_loop(root, i=0, row=None):
    if row == None:
        row = []

python - 变量不清除函数调用之间

1 个答案: