Python NumPy expand_dims方法用法及代码示例

Numpy 的 expand_dims(~) 方法返回一个新的 Numpy 数组，其中输入数组的维度已扩展。

1. a | array-like

输入数组。

2. axis | int

将添加新尺寸的轴。

一个新的 Numpy 数组，其中输入数组的维度已扩展。

假设我们有以下一维数组：

a = np.array([1,2,3,4])
a



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

我们可以通过两种方式将其扩展到二维：沿第 0 轴或第 1 轴添加新维度。

对于第 0 轴：

np.expand_dims(a, axis=0)



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

对于第 1 轴：

np.expand_dims(a, axis=1)



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

相关用法

注：本文由纯净天空筛选整理自Isshin Inada大神的英文原创作品 NumPy | expand_dims method。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。