python批量读取.mat文件并保存成.npy

import numpy as np
from scipy import io
import os

file_path = 'dataset/noise_FM'

# 批量处理mat数据,reshape为1000 x 1024,并存为.npy
def reshape_file(file_path):
    files = os.listdir(file_path)
    for file_origin in files:
        file = os.path.join(file_path, file_origin)
        data_mat = io.loadmat(file)
        data = data_mat['shiyu_pro'] #注意,通过io.loadmat加载的数据为字典格式
        data.reshape(1000, 1024)
        file_name = file_origin.split('.')[0]
        save_path = 'redata/noise_FM/' + file_name
        np.save(save_path, data)

# 批量加载.npy
def loadnpy(file_path):
    files = os.listdir(file_path)
    A[] = np.load(files)
    for file_origin in files:
        file = os.path.join(file_path, file_origin)
        data = np.load(file)
        A.append(data)

你可能感兴趣的:(深度学习从0到1,python,numpy,深度学习)