Python实现二维、三维、高维数组维度交换的几种方式

前言

在实际应用中，常常需要交换数组的维度以便于数据的处理，下面就整合了几种二维、三维和高维数组维度交换的方式，包括numpy.swapaxes、numpy.transpose、numpy.rollaxis、numpy.moveaxis函数等方式。其中，numpy.swapaxes 和 numpy.transpose 通常是最灵活和常用的

1 定义一个二维数组

以二维数组为例进行说明，打印结果为一个两行三列的数组

import numpy as np
arr = np.array([[1, 2, 3], [4, 5, 6]])  
print(arr)

# [[1 2 3]
#  [4 5 6]]  打印结果

2 numpy.transpose函数

打印结果为一个三行两列的数组，后面的结果类似

arr_trans = np.transpose(arr, (1, 0))  
print(arr_trans)

#[[1 4]
# [2 5]
# [3 6]]  打印结果

3 numpy.swapaxes函数

arr_swapped = np.swapaxes(arr, 1, 0)  
print(arr_swapped)

#[[1 4]
# [2 5]
# [3 6]]  打印结果

4 numpy.rollaxis函数

下面两种方式是将数组的轴进行滚动以完成数组维度的交换

arr_roll = np.rollaxis(arr, 1, 0) 
print(arr_roll)

arr_move = np.moveaxis(arr, 0, 1)  
print(arr_move)

#[[1 4]
# [2 5]
# [3 6]]  打印结果

后记

大家如果有问题需要交流或者有项目需要合作，可以加Q Q ：504156006详聊，加好友请留言“CSDN”，谢谢。