1. 程式人生 > >mat格式轉換成csv格式

mat格式轉換成csv格式

最近發現了一個超好的工具  也可以說是程式碼   值得推廣    能夠吧mat格式批量轉變成csv格式  超讚   網上的大部分都不是特別好,很多程式碼都不能很好的執行,現在有個Python版本的,分享給大家
# coding=utf-8
import scipy.io as sio
import pandas as pd
import os
def mat2csv():
    """ 將當前目錄下的data目錄下的 .mat 檔案轉換成多個 .csv檔案 :return: """
    curr_path = os.path.dirname(__file__)
    mat_data_path = os.path.join(curr_path, "data")
    csv_data_path = os.path.join(curr_path, "csv")
    if not os.path.exists(csv_data_path):
        os.makedirs(csv_data_path)
    if not os.path.exists(mat_data_path):
        os.makedirs(mat_data_path)
    file_list = os.listdir(mat_data_path)
    mat_list = [file_name for file_name in file_list if file_name.endswith(".mat")]
    print "find mat file : ", mat_list



    for mat_file in mat_list:
        file_path = os.path.join(mat_data_path, mat_file)
        mat_data = sio.loadmat(file_path)
        version = str(mat_data.get("__version__", "1.0")).replace(".", "_")
        for key in mat_data.iterkeys():
            if not str(key).startswith("__"):
                data = mat_data[key][:]
                print mat_file
                try:
                    dfdata = pd.DataFrame(data)
                except ValueError as e:
                    print e.message
                    continue
                csv_name = "_".join([mat_file.split(".")[0], key, version, '.csv'])
                csv_path = os.path.join(csv_data_path, csv_name)
                dfdata.to_csv(csv_path)
if __name__ == "__main__":
    mat2csv()