怎么給numpy.array增加維度

怎么給numpy.array增加維度？相信很多沒有經驗的人對此束手無策，為此本文總結了問題出現的原因和解決方法，通過這篇文章希望你能解決這個問題。

輸入：

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

輸出結果：

array([1, 2, 3])

輸入：

print(a[None])

輸出結果：

array([[1, 2, 3]])

輸入：

print(a[:,None])

輸出結果：

array([[1],               
       [2],               
       [3]])     

numpy數組的維度增減方法

使用np.expand_dims()為數組增加指定的軸,np.squeeze()將數組中的軸進行壓縮減小維度。

1.增加numpy array的維度

在操作數組情況下，需要按照某個軸將不同數組的維度對齊，這時候需要為數組添加維度(特別是將二維數組變成高維張量的情況下)。

numpy提供了expand_dims()函數來為數組增加維度：

import numpy as np
a = np.array([[1,2],[3,4]])
a.shape
print(a)
>>>
"""
(2L, 2L)
[[1 2]
 [3 4]]
"""
# 如果需要在數組上增加維度,輸入需要增添維度的軸即可，注意index從零還是
a_add_dimension = np.expand_dims(a,axis=0)
a_add_dimension.shape
>>> (1L, 2L, 2L)

a_add_dimension2 = np.expand_dims(a,axis=-1)
a_add_dimension2.shape
>>> (2L, 2L, 1L)

a_add_dimension3 = np.expand_dims(a,axis=1)
a_add_dimension3.shape
>>> (2L, 1L, 2L)

2.壓縮維度移除軸

在數組中會存在很多軸只有1維的情況，可以使用squeeze函數來壓縮冗余維度

b = np.array([[[[5],[6]],[[7],[8]]]])
b.shape
print(b)
>>>
"""
(1L, 2L, 2L, 1L)
array([[[[5],
         [6]],

        [[7],
         [8]]]])
"""
b_squeeze = b.squeeze()
b_squeeze.shape
>>>(2L, 2L)   #默認壓縮所有為1的維度

b_squeeze0 = b.squeeze(axis=0)   #調用array實例的方法
b_squeeze0.shape
>>>(2L, 2L, 1L)

b_squeeze3 = np.squeeze(b, axis=3)   #調用numpy的方法
b_squeeze3.shape
>>>(1L, 2L, 2L)

看完上述內容，你們掌握怎么給numpy.array增加維度的方法了嗎？如果還想學到更多技能或想了解更多相關內容，歡迎關注億速云行業資訊頻道，感謝各位的閱讀！