【python小练】0017-将xls文件内容写入xml文件中

第 0017 题： 将 第 0014 题中的 student.xls 文件中的内容写到 student.xml 文件中，如

下所示：

<?xml version="1.0" encoding="UTF-8"?>
<root>
<students>
<!--
    学生信息表
    "id" : [名字, 数学, 语文, 英文]
-->
{
    "" : ["张三", 150, 120, 100],
    "" : ["李四", 90, 99, 95],
    "" : ["王五", 60, 66, 68]
}
</students>
</root>

这里就要用上xml这个module啦，我打算用xml.dom.minidom来做

Code:

import xlrd
import xml.dom.minidom as md

def get_xls_data(filename):
     book = xlrd.open_workbook(filename)
     sheet = book.sheet_by_index(0)
     content = {}
     for i in range(sheet.nrows):
         content[i+1] = sheet.row_values(i)[1:]
     return content

def write_to_xml(xlscontent):

    xmlfile = md.Document()　　#创建新xml文件

    root = xmlfile.createElement('root')　　#创建节点
    students = xmlfile.createElement('students')　　#创建节点

    xmlfile.appendChild(root)　　#在文件中添加root节点
    root.appendChild(students)　　#在root下添加students节点

    comment = xmlfile.createComment('学生信息表 "id" : [名字, 数学, 语文, 英文]')　　#创建评论
    students.appendChild(comment)　　#在students标签下添加comment

    xmlcontent = xmlfile.createTextNode(str(xlscontent))　　　　#创建文本节点
    students.appendChild(xmlcontent)　　在students标签下添加文本内容

    with open('students.xml', 'wb') as f:
        f.write(xmlfile.toprettyxml(encoding = 'utf-8'))　　#写入文件

write_to_xml(get_xls_data('students.xls'))

Note:

1. get_xls_data(filename)函数从xls中读出内容

2. xml的相关操作并不难，見代码后注释

3. 注意通过get_xls_data()得到的content是dict，而createTextNode()方法要求创建内容对为str，用type(变量名)查看变量类型

获得的xml文件内容如下：

<?xml version="1.0" encoding="utf-8"?>
<root>
    <students>
        <!--学生信息表 "id" : [名字, 数学, 语文, 英文]-->
        {1: ['张三', '', '', ''], 2: ['李四', '', '', ''], 3: ['王五', '', '', '']}
    </students>
</root>

排成一行了，不知道能不能人为分行，不过对内容没影响