reshape - 检查 numpy 数组形状是否为 ()，如果是，则重塑为 (1,)

Question

我有一个函数可以将单个值作为 numpy 数组返回：

print(yhat, yhat.shape)
Output: 0.9337081 ()

如何检查 numpy 数组形状是否为 ()，如果是，则将其重塑为 (1,)？因此得到：

yhat = np.array([0.9337081])

标签： reshapenumpy-ndarray

Answer 1

()您已经通过打印验证 yhat 的形状是否等于yhat.shape。如果要重塑 numpy 数组，请使用reshape。

>>> yhat = np.array(0.9337081)
>>> yhat.shape
()
>>> yhat.shape == ()
True
>>> yhat = yhat.reshape(-1,)
>>> yhat.shape
(1,)
>>> yhat
array([0.9337081])