【游戲開發】Excel表格批量轉換成CSV的小工具

一、前言

　　在工作的過程中，我們有時可能會面臨將Excel表格轉換成CSV格式文件的需求。這尤其在游戲開發中體現的最為明顯，策划的數據文檔大多是一些Excel表格，且不說這些表格在游戲中讀取的速度，但就論占用內存來說，同樣的數據量Excel表格所占用的內存要遠遠大於CSV，因此將Excel轉換成CSV勢在必行。如果單單轉換一個Excel表格還好，直接另存為就搞定的，但是如何將一個文件下的N個Execl表格轉成CSV呢？今天馬三就來和大家一起用Python擼一個Excel表格批量轉換CSV的小工具——Xls2CSV。

二、准備開發環境

　　馬三在網上找到了一些現有的Excel批量轉換小工具，但是他們或多或少都有些不甚好用，里面的注釋又不全。馬三在它們的基礎上進行了優化，支持自定義輸入輸出路徑，解決了中文亂碼的問題，並且添加了較為全面的注釋。下面先簡單介紹一下開發環境：

• 轉化工具由Python2.7.x開發，依賴xlrd包（用於處理Excel表格）
• 首先安裝Python2.7.x並配置環境變量，這不必多說，網上有很多教程
• 然后安裝我們的依賴包xlrd，到https://pypi.python.org/pypi/xlrd 下載xlrd源碼包到本地，如圖1所示
• 在解壓后的源碼包內執行 python setup.py install ，安裝xlrd模塊
• 重啟命令行即可生效

　　

　　^{圖1：xlrd源碼包下載}

三、Xls2CSV腳本開發　　

　　開發環境准備好之后，我們就可以進行開發工作了。首先還是上一下整體的代碼，然后在解釋一下它的工作思路：

# -*- coding: utf-8 -*-  

import os,sys,inspect,re
import xdrlib,xlrd

# 防止中文亂碼
reload(sys)
sys.setdefaultencoding("utf-8")

# 分割符
C_SPACE = ","
# 結束符
C_END = "\n"
# 輸入路徑(存放xls文件的路徑）
IN_PATH = ""
# 輸出路徑(導出csv文件的路徑)
OUT_PATH = ""


# 讀取配置文件
def read_config():
    config_file = open("config.ini","r")
    cur_line = config_file.readline().rstrip("\r\n").split(',')
    global IN_PATH 
    IN_PATH = cur_line[1]
    cur_line = config_file.readline().rstrip("\r\n").split(',')
    global OUT_PATH 
    OUT_PATH = cur_line[1]

# 過濾路徑
def cur_file_dir(path):
    if os.path.isfile(path):
        path = os.path.dirname(path)
    print path
    return os.path.abspath(path)

# 搜索指定文件夾下面的文件
def find_file_by_pattern(pattern='.*', base=".", circle=True):  
    # 查找給定文件夾下面所有xls文件
    re_file = re.compile(pattern)
    # 第一次搜索的時候過濾下路徑，遞歸之后直接搜索base路徑即可
    if base == ".":
        base = cur_file_dir(IN_PATH)
    print u"開始搜索文件夾：",base

    # 存儲xls文件的列表
    final_file_list = []
    # 遍歷指定路徑下的文件
    cur_list = os.listdir(base)  
    for item in cur_list:
        # 忽略一些干擾的文件，如果你還有其他需要忽略的文件，直接在后面繼續添加即可
        if item == ".svn":
            continue
        # 拼接路徑
        full_path = os.path.join(base, item)
        # 忽略臨時文件
        if full_path.startswith("~"):
            continue
        # 篩選出xlsx\xls文件
        if full_path.endswith(".xlsx") or full_path.endswith(".xls"):
            print u"輸入文件:" + full_path
            bfile = os.path.isfile(item)
            if os.path.isfile(full_path):
                if re_file.search(full_path):
                    final_file_list.append(full_path)  
            else:
                final_file_list += find_file_by_pattern(pattern, full_path)

    # 返回文件列表
    return final_file_list


# 打開excel
def open_excel(file= 'file.xls'):
    try:
        data = xlrd.open_workbook(file)
        return data
    except Exception,e:
        print str(e)


#根據索引獲取Excel表格中的數據 參數:file：Excel文件路徑, colnameindex：表頭列名所在行的索引, by_index：表的索引
def excel_table_byindex(file='file.xls', colnameindex=0, by_index=0):
    data = open_excel(file)
    table = data.sheets()[by_index]
    nrows = table.nrows #行數
    ncols = table.ncols #列數
    rowlist = []
    
    '''開始讀取數據'''
    for rownum in range(colnameindex, nrows):
        rowdata = table.row_values(rownum)
        if rowdata:
            collist = []
            for i in range(ncols):
                collist.append(rowdata[i])
            rowlist.append(collist)
    return rowlist

#保存csv文件
def savaToCSV(_file, _list, _path):
    filename = ""
    content = ""
    #生成文件內容
    for collist in _list:
        for i in range(len(collist)):
            v = collist[i]
            vstr = ""
            # print k,v
            if isinstance(v, float) or isinstance(v, int):
                vstr = str(int(v))
            else:
                vstr = v
            if i > 0:
                content = content + C_SPACE
            content = content + vstr
        content = content + C_END

    #生成文件后綴 
    fname = os.path.basename(_file).split('.')
    filename = fname[0] + ".csv"

    #寫文件
    if len(filename)>0 and len(content)>0:
        filename = OUT_PATH + "/" + filename
        print u"輸出文件:" + filename
        file_object = open(filename, 'w')
        file_object.write(content)
        file_object.close()


def main():

    read_config()
    filelist = find_file_by_pattern()
    if len(filelist) > 0:
        path = ""
        #遍歷文件生成csv
        for file in filelist:
            datalist = excel_table_byindex(file, 0)
            if len(datalist)>0:
                savaToCSV(file, datalist, path)
    else:
        print u"沒有找到任何excel文件！"

if __name__=="__main__":
    main()

　　轉表工具的工作思路如下：

• 首先，從配置文件中讀取到輸入路徑和輸出路徑
• 接着，過濾路徑，並在輸入路徑下執行搜索，如果是xls文件的話，則把其路徑+文件名加入一個filelist列表中保存待用；如果是其他文件，則跳過；如果是目錄的話，則對該目錄也進行相同的搜索（即遞歸調用搜索函數）
• 然后，遍歷之前保存的filelist列表，取出xls文件的路徑，然后利用xlrd模塊打開Excel表格，並提取其中的信息
• 最后，根據之前的xls文件名，創建出對應的輸出路徑(輸出路徑+Execl文件名.csv)，然后將讀取出的數據，逐行寫入CSV文件並保存

　　具體每塊函數和代碼的作用，已經在注釋里面寫清了，直接看注釋就好。

四、Xls2CSV工具的使用

　　工具開發完了，如何使用呢？其實它使用起來很簡單，只需兩步操作：

• 打開 config.ini 配置文件，配置輸入路徑以及輸出路徑，將路徑替換成自己的就好 ，如圖2所示
• 執行 xlsx2csv.py 腳本進行自動導表，這一步最好在命令行下執行腳本，這樣如果有報錯信息的話也可以直觀地看到。如果雙擊執行的話，很有可能一閃而過，腳本報錯了也不知道

　　

　　^{圖2：配置輸入輸出路徑}

五、其他優秀的轉表工具 　　

　　上面我們寫的Excel批量轉換工具其實還是太簡陋了，tolua的開發者已經為廣大的Unity開發人員制作了一款可以將Excel表格數據導出為Lua table、csv、json形式的工具，兼帶數據檢查功能以及導出、導入MySQL數據庫功能。除此之外，還支持GUI界面等很多實用的功能，大家感興趣的話可以到Github去查看該項目的具體內容：https://github.com/zhangqi-ulua/XlsxToLua

六、總結

　　通過本篇博客，我們一起學習了如何使用Python制作一款批量轉換Excel表格為CSV文件的小工具，從而提升我們的工作效率。因為本工具並沒有經過大量項目和數據的測試，所以很有可能存在未知的bug，因此如果大家在商業項目中應用的話，最好還是使用上面所說的成熟的tolua中的轉表工具，會更穩定一些。本篇博客只是為大家拋磚引玉，提供一種Excel批量轉換的思路。

　　本篇博客中的所有代碼已經托管到Github，開源地址：https://github.com/XINCGer/Unity3DTraining/tree/master/XlsxTools/xls2csv  歡迎fork!

 

 

作者：馬三小伙兒
出處：http://www.cnblogs.com/msxh/p/7858346.html
請尊重別人的勞動成果，讓分享成為一種美德，歡迎轉載。另外，文章在表述和代碼方面如有不妥之處，歡迎批評指正。留下你的腳印，歡迎評論！