mat矩陣和npy矩陣實現互相轉換(python和matlab)

Posted on by

mat矩陣和npy矩陣互相轉換

numpy.narray矩陣保存為mat文件

import numpy as np
import scipy.io as io
mat_path = 'your_mat_save_path'
mat = np.zeros([4, 20])
io.savemat(mat_path, {'name': mat})

註意這裡的mat是numpy類型的

讀取mat文件

import numpy as np
from scipy import io
mat = io.loadmat('yourfile.mat')
# 如果報錯:Please use HDF reader for matlab v7.3 files
# 改為下一種方式讀取
import h5py
mat = h5py.File('yourfile.mat')
# mat文件裡可能有多個cell，各對應著一個dataset
# 可以用keys方法查看cell的名字, 現在要用list(mat.keys()),
# 另外，讀取要用data = mat.get('名字'), 然後可以再用Numpy轉為array
print(mat.keys())
# 可以用values方法查看各個cell的信息
print(mat.values())
# 可以用shape查看維度信息
print(mat['your_dataset_name'].shape)
# 註意，這裡看到的shape信息與你在matlab打開的不同
# 這裡的矩陣是matlab打開時矩陣的轉置
# 所以，我們需要將它轉置回來
mat_t = np.transpose(mat['your_dataset_name'])
# mat_t 是numpy.ndarray格式
# 再將其存為npy格式文件
np.save('yourfile.npy', mat_t)

npy文件與mat文件的保存與讀取

除瞭常用的csv文件和excel文件之外，我們還可以通過PY把數據保存文npy文件格式和mat文件格式。

1. npy文件

npy即numpy對應的文件格式，關於其保存使用的是np.save()方法，其讀取使用的是np.load()方法。 

具體示例如下：

import numpy as np
a = np.mat('1, 2, 3;4, 5, 6')
print(a)
print(type(a))
print("=================================")
b = np.array([[1, 2, 3], [4, 5, 6]])
print(b)
print(type(b))

              

保存文件：

如圖，矩陣和numpy數組都支持以npy文件類型保存。

np.save('a.npy', a)
np.save('b.npy', b)

                    

讀取文件

data1 = np.load('a.npy')
data2 = np.load('b.npy')
print(data1)
print(type(data1))
print("=================================")
print(data2)
print(type(data2))

               

如圖npy數據被成功讀取，且都是numpy數組數據類型。

2. mat文件

保存為mat文件依賴於scipy庫中的scipy.io.savemat()方法，讀取則需要用到scipy.io.loadmat()方法。

保存時，不僅僅需要傳入變量，還需要將該變量的類型一並以字典的形式傳入，一樣支持numpy數組和矩陣。

具體示例如下：

import numpy as np
from scipy import io
a = np.mat('1, 2, 3;4, 5, 6')
print(a)
print(type(a))
print("=================================")
b = np.array([[1, 2, 3], [4, 5, 6]])
print(b)
print(type(b))
io.savemat('a.mat', {'matrix': a})
io.savemat('b.mat', {'array': b})

讀取數據

data1 = io.loadmat('a.mat')
print(data1)
print(type(data1))
print("=================================")
data2 = io.loadmat('b.mat')
print(data2)
print(type(data2))

如圖，數據成功被讀取。但是讀取的結果是一個字典，如果需要進一步讀取到數據，則需要根據鍵名將其取出：

print(data1['matrix'])
print(type(data1['matrix']))
print("=================================")
print(data2['array'])
print(type(data2['array']))

               

取出時的鍵與存儲時的變量類型有關，取出的數據都是numpy數組，不再有矩陣類型。

以上為個人經驗，希望能給大傢一個參考，也希望大傢多多支持WalkonNet。

推薦閱讀：