报错
ValueError: Expected 2D array, got 1D array instead:
报错分析
期待二维数组,却得到的是一维数组
报错解决
简单!在以为数组外面套一个[]
就可以了!
# 原报错代码
for i in vec:
np.mean(i) # 得到的是一维数组
更改为下面的代码就好:
for i in vec:
[np.mean(i)] # 得到的是二维数组
知识补充
np.stack()
np.stack()用于堆叠数组,默认axis=0
,表示将多个数组在第0维
上进行堆叠,变为一个大数组。
例如:
vec = [array([1, 2, 3, 4, 5, 6], dtype=float32),
array([10, 11, 12, 13, 14, 15], dtype=float32),
array([20, 21, 22, 23, 24, 25], dtype=float32),
]
堆叠:
np.stack(vec)
得到的结果是:
array([[1, 2, 3, 4, 5, 6],
[10, 11, 12, 13, 14, 15],
[20, 21, 22, 23, 24, 25]], dtype=float32)