将float转换为ndarray或反之亦然

Question

我想要计算平均绝对误差，但是当我执行以下打印语句时

print('MAE:  %5.2f kcal/mol'%np.abs(output - TZ).mean(axis=0))

我收到以下错误

TypeError: float argument required, not numpy.ndarray

其他信息：

output = 

[-0.01200975 -0.01200975 -0.01200975 ..., -0.01042357 -0.01200975
 -0.01042357]

TZ =

[[ 2.43307638]
 [ 1.981444  ]
 [ 0.25338271]
 ..., 
 [-0.18407178]
 [-0.734635  ]
 [ 1.724509  ]]

我是python的新手，有人能告诉我如何将TZ转换为float，反之亦然吗？

Answer 1

我想我找到了答案。

我将输出重新整形为output.reshape（output.shape [0]，1），现在两个变量的格式相同。