报错与解决 | ValueError: Expected 2D array, got 1D array instead:

报错

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)

你可能感兴趣的:(报错与解决,Python,python,numpy)