當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python Numpy.expand_dims()用法及代碼示例


借助Numpy.expand_dims()方法,我們可以獲得數組的擴展維Numpy.expand_dims()方法。

用法: Numpy.expand_dims()

返回:返回擴展數組。


範例1:
在這個例子中,我們可以看到Numpy.expand_dims()方法,我們可以使用此方法獲取擴展數組。

# import numpy 
import numpy as np 
  
# using Numpy.expand_dims() method 
gfg = np.array([1, 2]) 
print(gfg.shape) 
  
gfg = np.expand_dims(gfg, axis = 0) 
print(gfg.shape)

輸出:

(2, )
(1, 2)

範例2:

# import numpy 
import numpy as np 
  
# using Numpy.expand_dims() method 
gfg = np.array([[1, 2], [7, 8]]) 
print(gfg.shape) 
  
gfg = np.expand_dims(gfg, axis = 0) 
print(gfg.shape)

輸出:

(2, 2)
(1, 2, 2)



相關用法


注:本文由純淨天空篩選整理自Jitender_1998大神的英文原創作品 Python | Numpy.expand_dims() method。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。