np.expand_dims功能

np.expand_dims 是一个 numpy 库中用于扩充数组维度的函数。它接收一个数组 (array) 和一个 axis 参数，并在指定的 axis 上插入一个新的维度。例如，当 axis=0 时，np.expand_dims(arr, axis=0) 会在 arr 的最前面插入一个新的维度，使得输出的数组的 shape 变成 (1, arr.shape)。

相关问题

解释一下np.expand_dims

np.expand_dims是NumPy库中的一个函数，用于在指定的位置插入新的维度。它的作用是扩展数组的维度，可以在数组的任意位置插入新的维度。

具体来说，np.expand_dims函数接受两个参数：第一个参数是要扩展维度的数组，第二个参数是要插入的位置。插入位置可以是一个整数或一个元组。

如果插入位置是一个整数，则在该整数所指定的位置插入新的维度。例如，对于一个形状为(3, 4)的二维数组，使用np.expand_dims(arr, 0)将在第0个位置插入新的维度，得到一个形状为(1, 3, 4)的三维数组。

如果插入位置是一个元组，则在元组中指定的多个位置同时插入新的维度。例如，对于一个形状为(3, 4)的二维数组，使用np.expand_dims(arr, (0, 2))将在第0个和第2个位置同时插入新的维度，得到一个形状为(1, 3, 1, 4)的四维数组。

np.expand_dims

The numpy function np.expand_dims() is used to add an extra dimension to an existing numpy array.

Syntax:

np.expand_dims(arr, axis)

Parameters:
- arr: the numpy array to be expanded
- axis: the axis along which the new dimension will be added. If axis is not specified, a new dimension will be added at the beginning of the array.

Returns:
- A new numpy array with the same data as the original array, but with an additional dimension inserted.

Example:

import numpy as np

arr = np.array([1, 2, 3])
print(arr.shape)  # Output: (3,)

# Add a new dimension to the array
new_arr = np.expand_dims(arr, axis=0)
print(new_arr.shape)  # Output: (1, 3)

# Add a new dimension to the array at the end
new_arr2 = np.expand_dims(arr, axis=1)
print(new_arr2.shape)  # Output: (3, 1)

In this example, we first define a numpy array `arr` with shape (3,). We then use np.expand_dims() to add a new dimension to the array at axis=0 and axis=1. The resulting arrays are `new_arr` with shape (1, 3) and `new_arr2` with shape (3, 1).