温馨提示×

numpy三维数组转置的方法是什么

小亿
153
2023-09-29 10:30:29
栏目: 编程语言

在numpy中,可以使用numpy.transpose()函数来进行三维数组的转置。该函数的语法如下:

numpy.transpose(arr, axes)

其中,arr是要转置的数组,axes是一个整数元组,用于指定转置后的维度顺序。如果不指定axes参数,则默认为None,此时将会翻转数组的所有维度。

以下是一个示例:

import numpy as np
# 创建一个三维数组
arr = np.array([[[1, 2], [3, 4]], [[5, 6], [7, 8]]])
print("原数组:")
print(arr)
# 转置数组
transposed_arr = np.transpose(arr, axes=(2, 0, 1))
print("转置后的数组:")
print(transposed_arr)

输出结果为:

原数组:
[[[1 2]
[3 4]]]
[[[5 6]
[7 8]]]
转置后的数组:
[[[1 3]
[5 7]]
[[2 4]
[6 8]]]

上述示例中,原数组的维度顺序是(0, 1, 2),通过指定axes=(2, 0, 1),将其转置为(2, 0, 1)的顺序。

1