当前位置: 首页>>代码示例 >>用法及示例精选 >>正文

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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。