Проверьте numpy массив массива () и, если это так, измените его на (1,)

Question

У меня есть функция, которая может возвращать одно значение в виде numpy массива:

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

Как проверить, что numpy форма массива is () и, если это так, изменить его на (1,) ? Итак, в результате получим:

yhat = np.array([0.9337081])

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])