numpy如何交换数组维度,需要具体代码示例
numpy是Python中一个强大的数值计算库,它提供了许多方便的函数和方法来处理数组。在numpy中,我们可以轻松地交换数组的维度。本文将介绍如何使用numpy库中的函数来实现数组维度的交换,并提供具体的代码示例。
在numpy中,交换数组维度可以使用transpose()和swapaxes()两个函数。这两个函数用于对N维数组进行维度变换,可以很方便地交换维度顺序。
首先,我们来看transpose()函数。这个函数可以接受一个由维度索引组成的元组作为输入参数,用于指定维度的交换顺序。例如,我们有一个二维数组arr,想要交换其维度,可以使用transpose()函数如下:
import numpy as np
arr = np.array([[1, 2], [3, 4]])
print("原数组:
", arr)
# 交换维度
arr_transposed = np.transpose(arr)
print("交换维度后的数组:
", arr_transposed)




