numpy expand_dims - Python

在NumPy中，expand_dims函数是用于在指定的轴位置添加新的维度。这个函数将数组的维度增加1，并且这个新的维度被放在 指定的轴位置上，并且在其他轴上复制原来的数据。下面是这个函数的语法：

numpy.expand_dims(array, axis)

• array：需要增加一个新的维度的数组
• axis：指定新维度将插入的位置

返回值是一个新的具有插入新维度的数组。

示例

这里有一个简单的示例，展示了如何使用expand_dims来增加一个新的维度：

import numpy as np
  
# 创建一个一维数组
arr = np.array([1, 2, 3])
print("arr:")
print(arr)
  
# 在第二个位置添加一个新的维度
new_arr = np.expand_dims(arr, axis=1)
print("\nnew_arr:")
print(new_arr)

输出结果为：

arr:
[1 2 3]

new_arr:
[[1]
 [2]
 [3]]

在这个例子中，我们首先创建了一个一维数组[1, 2, 3]。然后我们使用expand_dims函数在第二个位置添加了一个新的维度。这个新维度的维度为1，所以整个数组变成了一个二维数组，其中有三个行和一个列。

总结

在NumPy中，使用expand_dims函数可以很方便地在指定的轴位置添加新的维度。这对于一些需要处理多维数组的数据分析任务来说非常有用。