在 NumPy 数组中插入一个新轴

原文:https://www . geeksforgeeks . org/insert-a-new-axis-in-a-numpy-array/

这篇文章讨论了在 NumPy 中增加数组维数的方法。NumPy 为我们提供了两种不同的内置函数来增加数组的维数,即

1D array will become 2D array
2D array will become 3D array
3D array will become 4D array
4D array will become 5D array

方法一:使用 numpy.newaxis() 第一种方法是使用 numpy.newaxis 对象。此对象相当于在声明数组时使用无作为参数。诀窍是在要添加新轴的索引位置使用 numpy.newaxis 对象作为参数。 例:

Python 3

import numpy as np

arr = np.arange(5*5).reshape(5, 5)
print(arr.shape)

# promoting 2D array to a 5D array
# arr[None, ..., None, None]
arr_5D = arr[np.newaxis, ..., np.newaxis, np.newaxis]

print(arr_5D.shape)