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参数来控制转置后数组的维度顺序。它是处理多维数组时非常方便实用的工具。
极客教程