Python批量將csv文件轉化成xml文件的實例

更新時間：2021年05月10日 16:24:04 作者：葉庭云

將 csv 格式轉換成xml格式有許多方法，可以用數據庫的方式，也有許多軟件可以將 csv 轉換成xml。但是比較麻煩，本文利用 Python 一鍵批量將 csv 文件轉化成 xml 文件。

一、前言

逗號分隔值（Comma-Separated Values，CSV，有時也稱為字符分隔值，因為分隔字符也可以不是逗號），其文件以純文本形式存儲表格數據（數字和文本）。純文本意味著該文件是一個字符序列，不含必須像二進制數字那樣被解讀的數據。CSV文件由任意數目的記錄組成，記錄間以某種換行符分隔；每條記錄由字段組成，字段間的分隔符是其它字符或字符串，最常見的是逗號或制表符。通常，所有記錄都有完全相同的字段序列，通常都是純文本文件。

可擴展標記語言，標準通用標記語言的子集，簡稱XML。是一種用于標記電子文件使其具有結構性的標記語言。在電子計算機中，標記指計算機所能理解的信息符號，通過此種標記，計算機之間可以處理包含各種的信息比如文章等。它可以用來標記數據、定義數據類型，是一種允許用戶對自己的標記語言進行定義的源語言。它非常適合萬維網傳輸，提供統(tǒng)一的方法來描述和交換獨立于應用程序或供應商的結構化數據。

二、Python代碼實現

導入用到的庫

from xml.etree.ElementTree import Element, ElementTree
import csv
from pathlib import Path
import os

創(chuàng)建一個文件夾，用來保存轉換后的xml文件

# 創(chuàng)建一個文件夾  用來保存轉換后的xml文件
path = os.path.join('xml_file')
if not os.path.exists(path):
    os.mkdir(path)

獲取所有待轉換的csv文件

# 獲取所有待轉換的csv文件  返回列表
def list_csv():
    file_path = input('請輸入你存放csv文件的路徑：')
    p = Path(file_path)
    csv_files = p.glob('**/*.csv')
    csv_files = [str(csv_file) for csv_file in csv_files]
    return csv_files

將 csv 文件轉換為 xml 文件

# 將csv文件轉換為xml
def csv_to_xml(file_name):
    print(file_name)
    with open(file_name, 'r', encoding='utf-8') as f:   # 讀取csv文件
        reader = csv.reader(f)
        header = next(reader)   # 跳過表頭
        root = Element('Datas')
        print('root', len(root))
        # 處理轉換
        for row in reader:
            erow = Element('row')
            root.append(erow)
            for tag, text in zip(header, row):
                e = Element(tag)
                e.text = text
                erow.append(e)
    beatau(root)
    return ElementTree(root)


def beatau(e, level=0):
    if len(e) > 0:
        e.text = '\n' + '\t' * (level + 1)
        child = None
        for child in e:
            beatau(child, level + 1)
        child.tail = child.tail[:-1]
    e.tail = '\n' + '\t' * level

主函數調用

if __name__ == '__main__':
    csv_list = list_csv()
    print(csv_list)
    for index_, item in enumerate(csv_list, start=1):
        print(index_, item)
        et = csv_to_xml(item)
        et.write(path + '/' + 'test_{}.xml'.format(index_), encoding='utf-8')

結果如下：