前言
在实际应用中,常常需要交换数组的维度以便于数据的处理,下面就整合了几种二维、三维和高维数组维度交换的方式,包括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]] 打印结果
后记
大家如果有地信遥感方面的问题需要请教或者有项目需要合作,可以在闲 鱼软件的用户中搜索:遥感GIS工作室,请认准头像,谢谢。