numpy.transpose
在NumPy中,transpose
函数用于交换数组的维度。通过transpose
函数可以方便地对数组进行转置操作,是NumPy中非常常用的函数之一。
语法
该函数的语法如下:
numpy.transpose(array, axes)
参数说明:
array
:要进行转置操作的数组。axes
:可选参数,用于指定转置后数组的维度顺序。默认值为None
,表示按照原数组的索引顺序进行转置。
示例代码1
下面先看一个简单的示例代码:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
transposed_arr = np.transpose(arr)
print("原数组:")
print(arr)
print("转置后的数组:")
print(transposed_arr)
运行结果为:
原数组:
[[1 2 3]
[4 5 6]]
转置后的数组:
[[1 4]
[2 5]
[3 6]]
在这个示例中,原始数组arr
的形状是(2, 3)
,转置后的数组transposed_arr
的形状变为(3, 2)
,即原数组的行和列交换。
示例代码2
下面再看一个带有axes
参数的示例代码:
import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])
transposed_arr = np.transpose(arr, axes=(1, 0))
print("原数组:")
print(arr)
print("按指定顺序转置后的数组:")
print(transposed_arr)
运行结果为:
原数组:
[[1 2 3]
[4 5 6]]
按指定顺序转置后的数组:
[[1 4]
[2 5]
[3 6]]
这里指定转置后数组的维度顺序为(1, 0)
,与上一个示例相同,因此结果也是相同的。
总结一下,numpy.transpose
函数可以帮助我们在NumPy中进行数组的转置操作,并且可以通过传入axes
参数来控制转置后数组的维度顺序。它是处理多维数组时非常方便实用的工具。