Python 保存向量和矩阵到文件：scipy、numpy、h5py、pickle 方法比较

方法一：

使用 scipy 库中的 io 模块中的 savemat() 函数将向量和矩阵保存到 .mat 文件中。

代码示例：

import scipy.io as sio
import numpy as np

# 创建向量和矩阵
vector = np.array([1, 2, 3])
matrix = np.array([[1, 2], [3, 4]])

# 保存到 .mat 文件中
sio.savemat('data.mat', {'vector': vector, 'matrix': matrix})

方法二：

使用 numpy 库中的 savez() 函数将向量和矩阵保存到 .npz 文件中，再使用 load() 函数加载数据。

代码示例：

import numpy as np

# 创建向量和矩阵
vector = np.array([1, 2, 3])
matrix = np.array([[1, 2], [3, 4]])

# 保存到 .npz 文件中
np.savez('data.npz', vector=vector, matrix=matrix)

# 加载数据
data = np.load('data.npz')
vector = data['vector']
matrix = data['matrix']

方法三：

使用 h5py 库中的 File() 函数将向量和矩阵保存到 .h5 文件中，再使用 get() 函数加载数据。

代码示例：

import h5py
import numpy as np

# 创建向量和矩阵
vector = np.array([1, 2, 3])
matrix = np.array([[1, 2], [3, 4]])

# 保存到 .h5 文件中
with h5py.File('data.h5', 'w') as f:
    f.create_dataset('vector', data=vector)
    f.create_dataset('matrix', data=matrix)

# 加载数据
with h5py.File('data.h5', 'r') as f:
    vector = f['vector'][:]
    matrix = f['matrix'][:]

方法四：

使用 pickle 库将向量和矩阵保存到 .pkl 文件中，再使用 load() 函数加载数据。

代码示例：

import pickle
import numpy as np

# 创建向量和矩阵
vector = np.array([1, 2, 3])
matrix = np.array([[1, 2], [3, 4]])

# 保存到 .pkl 文件中
with open('data.pkl', 'wb') as f:
    pickle.dump({'vector': vector, 'matrix': matrix}, f)

# 加载数据
with open('data.pkl', 'rb') as f:
    data = pickle.load(f)
    vector = data['vector']
    matrix = data['matrix']

以上四种方法都可以将向量和矩阵保存到文件中，根据实际情况选择合适的方法。