要交换NumPy数组的维度,可以使用切片(slicing)方法。下面是一个示例代码:
import numpy as np
# 创建一个2x3的二维数组
arr = np.array([[1, 2, 3],
[4, 5, 6]])
# 使用切片交换数组的维度
arr_swap = arr[:, ::-1]
# 打印交换后的数组
print(arr_swap)
输出结果:
[[3 2 1]
[6 5 4]]
在上面的代码中,我们首先创建了一个2x3的二维数组arr
。然后,我们使用切片[:, ::-1]
来交换数组的维度,其中:
表示选取所有的行,::-1
表示逆序选择所有的列。最后,我们将交换后的数组赋值给arr_swap
并打印出来。
通过这种方法,我们可以很容易地交换NumPy数组的维度。
下一篇:布尔情况