将 mat 文件转换为 pandas 数据帧

Question

我想将 this file 转换为 pandas 数据帧。

import pandas as pd
import scipy.io
mat = scipy.io.loadmat('cardio.mat')
cardio_df = pd.DataFrame(mat)

我收到此错误： Exception: Data must be 1-dimensional

Answer 1

似乎 mat 是一个包含 X 形状 (1831, 21)、y 形状 (1831, 1) 和一些元数据的字典。假设 X 是数据，y 是相同的标签，你可以用 np.hstack 水平堆叠它们并将它们加载到 pandas:

In [1755]: mat = scipy.io.loadmat('cardio.mat')

In [1758]: cardio_df = pd.DataFrame(np.hstack((mat['X'], mat['y'])))

In [1759]: cardio_df.head()
Out[1759]: 
         0         1         2         3         4         5         6   \
0  0.004912  0.693191 -0.203640  0.595322  0.353190 -0.061401 -0.278295   
1  0.110729 -0.079903 -0.203640  1.268942  0.396246 -0.061401 -0.278295   
2  0.216546 -0.272445 -0.203640  1.050988  0.148753 -0.061401 -0.278295   
3  0.004912  0.727346 -0.203640  1.212171 -0.683598 -0.061401 -0.278295   
4 -0.100905  0.363595  1.321366  1.027120  0.141359 -0.061401 -0.278295   

In [1760]: cardio_df.shape
Out[1760]: (1831, 22)

将 mat 文件转换为 pandas 数据帧

Convert mat file to pandas dataframe

python

pandas

dataframe

mat