数据处理

2023年春季PRP项目中遇到的函数整理。

.mat文件读取/保存

• Scipy.io 方法（处理一般的.mat文件）

  scio.loadmat("name.mat") data = np.array(matName['data'], dtype=float)

scio.savemat("data.mat",{"matName":data})

• h5py方法 （处理 save -v7.3 形式保存的.mat文件）

data = h5py.File("DO_1990.mat", "r")
diss_oxygen = data['do1']
diss_oxygen = torch.tensor(diss_oxygen)

numpy操作

• y = np.full_like(x, -1) 创建一个与x大小相同的空数组并初始化

• array_np.T[0] array_np[,:0] 获取首列

• array_np[0] 获取首行

• 一维数组特征
  np.min(array_np) np.max(array_np) np.average(array_np)
np.cumsum(array_np) 累加 (a[i] -> s[i])

• array_np.astype(int) 强制类型转换

• array_np.shape 获取形状

torch操作

• indices = torch.where(array = key) 返回符合条件的索引元组

• unique_list = torch.unique(torch.tensor(data)) 一维张量去重