不从主返回的值

Question

我有以下代码，我希望将输入文件和输出文件拉入到其他函数中，但是，这些代码似乎没有返回。仍然很新，所以如果这很简单，请道歉。# PYTHON 3.76 ONLY# Version 0.0.1import xml.etree.cElementTree as etimport pandas as pdimport sys, getoptdef main(argv):    inputfile = ''    outputfile = ''    try:        opts, args = getopt.getopt(argv,"hi:o:",["ifile=","ofile="])    except getopt.GetoptError:        print ('test.py -i <inputfile> -o <outputfile>')        sys.exit(2)    for opt, arg in opts:        if opt == '-h':            print ('test.py -i <inputfile> -o <outputfile>')            sys.exit()        elif opt in ("-i", "--ifile"):            inputfile = arg        elif opt in ("-o", "--ofile"):            outputfile = arg    print ('Input file is "', inputfile)    print ('Output file is "', outputfile)    return inputfile, outputfileif __name__ == "__main__":    main(sys.argv[1:])# convert XML to dataframedef xml2df(xml_data):    tree = et.parse(xml_data)    print (tree.getroot())    root = tree.getroot()    print ("tag=%s, attrib=%s" % (root.tag, root.attrib))    #iterate over each value for room and each user and add to rows    rows = []    for child in root.iter('rooms'):        roomId, roomTitle = 'id', 'ttl'    for it in child:        if it.tag == 'room':            roomId = it.findtext('roomID')            roomTitle = it.findtext('roomTitle')            roomStatus = it.findtext('status')            isAnonymous = it.findtext('isAnonymous')返回错误回溯（最近调用最后）：文件“parse_pChatDump.py”，第 63 行，在 df = xml2df（输入文件） 名称错误：未定义名称“输入文件”

Answer 1

您在方法中创建和返回它，但是您不将它们存储在脚本开始的变量中，因此以后无法访问它们，请执行：inputfileoutputfilemain

def main(argv):

    inputfile = ''

    outputfile = ''

    # ...

    return inputfile, outputfile

def xml2df(xml_data):

    # ...

    return df

if __name__ == "__main__":

    inputfile, outputfile = main(sys.argv[1:])

    df = xml2df(inputfile)

    df.to_csv(outputfile + ".csv", sep=',', index=False)