不知道是pandas的bug还是特性,使用pandas确实可以装载高维的list数据,而直接使用高维的np.ndarray会报错。
# 使用pandas装载3维数据 import pandas as pd import numpy as np data = np.ones([2, 2, 2]) data = data[:, np.newaxis, :, :] # 必须插入第二维作为pandas的column data = data.tolist() df = pd.DataFrame() df['data'] = data print(df)
打印结果:
data
0 [[[1.0, 1.0], [1.0, 1.0]]]
1 [[[1.0, 1.0], [1.0, 1.0]]]