作用：從數(shù)組的形狀中刪除單維條目，即把shape中為1的維度去掉

例子：

import numpy as npa = np.array([[1], [2], [3]]) print(a) print(a.shape)

輸出：

[[1]
?[2]
?[3]]
shape:(3, 1)

應(yīng)用squeeze()后：

a1 = np.squeeze(a) print(a1) print(a1.shape)

[1 2 3]
shape:(3,)

應(yīng)用：

在預(yù)測(cè)分析中用于處理預(yù)測(cè)數(shù)組和真實(shí)值數(shù)組以方便計(jì)算預(yù)測(cè)值與真實(shí)值之間的誤差：

predictions = np.array(predictions).squeeze() labels = np.array(labels).squeeze() rmse = np.sqrt(((predictions - labels) ** 2).mean(axis=O))

?

總結(jié)

以上是生活随笔為你收集整理的np.squeeze()的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。