讀取xml文件，寫入excel

　　在上一篇 Python寫xml文件 已經將所有訂單寫入xml文件，這一篇我們把xml文件中的內容讀出來，寫入excel文件。

　　輸入xml格式：

<?xml version="1.0" encoding="utf-8"?>
<orderlist>
    <order>
        <customer>姓名1</customer>
        <phone>123456</phone>
        <address>成都</address>
        <count>2</count>
    </order>
    <order>
        <customer>姓名2</customer>
        <phone>234567</phone>
        <address>成都</address>
        <count>5</count>
    </order>
    <order>
        <customer>姓名3</customer>
        <phone>345678</phone>
        <address>成都</address>
        <count>1</count>
    </order>
</orderlist>

　　輸出excel格式（這里按點餐次數進行的降序排序）：

　　工具庫選擇：beautifulsoup+lxml用於處理xml文件（當然也可直接使用lxml）， xlsxwriter用於寫excel文件。

　　思路也比較簡單：找出xml文件中的每一個<order>，然后進行降序排序，再將每一個order信息寫入excel文件，每個order一行。

　　代碼如下：

# coding: utf-8

import bs4
import xlsxwriter

# 讀取xml文件，寫入excel
def xmlToExcel(file_xml, file_excel):
    # 打開xml文件，並以此創建一個bs對象
    xml = open(file_xml, 'r')
    doc = bs4.BeautifulSoup(xml, 'xml')

    # 創建一個excel文件，並添加一個sheet，命名為orders
    workbook = xlsxwriter.Workbook(file_excel)
    sheet = workbook.add_worksheet('orders')

    # 設置粗體
    bold = workbook.add_format({'bold': True})

    # 先在第一行寫標題，用粗體
    sheet.write('A1', u'姓名', bold)
    sheet.write('B1', u'電話', bold)
    sheet.write('C1', u'點餐次數', bold)
    sheet.write('D1', u'地址', bold)

    # 篩選出所有的<order>，這里使用的是CSS選擇器
    order = doc.select('order')

    # 以每個order的count元素，對order進行降序排序
    sort_key = lambda a: int(a.count.text)
    order.sort(key=sort_key, reverse=True)

    # 行號，具體訂單信息從第二行開始
    row = 2
    # 將每一個訂單寫入excel
    for x in order:
        # 提取出具體信息
        name = x.customer.text
        phone = x.phone.text
        cnt = x.count.text
        addr = x.address.text

        # 將具體信息寫入excel
        sheet.write('A%d' % row, name)
        sheet.write('B%d' % row, phone)
        sheet.write('C%d' % row, cnt)
        sheet.write('D%d' % row, addr)

        row += 1

    # 關閉文件
    xml.close()
    workbook.close()

# 測試代碼
if __name__ == '__main__':
    file1 = 'hh.xml'
    file2 = 'hehe.xlsx'

    xmlToExcel(file1, file2)