1. 程式人生 > >【問題探究】numpy.squeeze()的輸入問題

【問題探究】numpy.squeeze()的輸入問題

前言:

{

    這篇部落格只介紹一個小問題,沒有經過多少檢索。

}

 

正文:

{

    numpy.squeeze(a, b)[1]可以把a的shape中b位置的1刪除。

#程式碼1
a = []
a.append([[1,2]])
b = np.squeeze(a, 0)

    我原本以為,程式碼1執行完後"b=[[1,2]]",但實際"b=[1,2]"。

    程式碼2則符合預期。

#程式碼2
a = []
a.append([[1,2]])
b = np.squeeze(np.array(a), 0)

    [1]中也沒有明確地介紹此問題,只說輸入型別是array_like,但是[2]中說array_like包括list。

}

 

結語:

{

    總之知道解決辦法了,雖然具體原因我也沒時間深究。

    參考資料:

    {

        [1]https://docs.scipy.org/doc/numpy-1.13.0/reference/generated/numpy.squeeze.html

        [2]https://docs.scipy.org/doc/numpy-1.13.0/glossary.html#term-array-like

    }

}