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

【问题标题】：Check numpy array shape is () and if so reshape to (1,)检查 numpy 数组形状是否为 ()，如果是，则重塑为 (1,)
【发布时间】：2020-02-26 18:20:12
【问题描述】：

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

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

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

yhat = np.array([0.9337081])

【问题讨论】：

标签： reshape numpy-ndarray

【解决方案1】：

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

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

【讨论】：