从python中的.mat文件访问指定的列

如何解决从python中的.mat文件访问指定的列

我总共有 61225 个矩阵，每个矩阵都有相同的变量名和不同的数据。我只对具有变量名称 Pose_Para 的文件感兴趣，这是一个 1x7 文件，我只对第 2、3 和 4 列感兴趣。我的问题是如何仅访问这些列才能获得大小为 [61225 x 3] 的矩阵？

这是我到现在为止所做的。

import glob
from scipy.io import loadmat
    
entries = glob.glob('300W_LP/*/*.mat',recursive=True)
print(len(entries))
for entry in entries:
    para = loadmat(entry,variable_names = ("Pose_Para",))["Pose_Para"]

解决方法

我假设 para = loadmat(entry,variable_names = ("Pose_Para",))["Pose_Para"] 导致 para 是一个 7 元素的 numpy 数组。如果是这种情况，您可以这样做：

import numpy as np

# ...file handling...
allparas = np.empty((len(entries),3))
for ctr,entry in enumerate(entries):
    allparas[ctr,:] = loadmat(entry,))["Pose_Para"][1:3]

请注意，您可能需要在 squeeze_me = True 调用中添加 loadmat。我不确定 loadmat 是否需要它来生成一个 7 元素的 1D Numpy 数组，而不是 1x7 2D 数组。

您还可以使用列表推导式将 for 循环编写为单行式：

allparas = np.array([loadmat(entry,variable_names= ("Pose_Para",))["Pose_Para"][1:3] for entry in entries])