NumPy库学习之transpose函数

NumPy库学习之transpose函数

一、简介

numpy.transpose 是 NumPy 库中的一个函数,用于对数组的轴进行置换。这个函数可以重新排列多维数组的维度,使得数组的各个维度可以按照指定的顺序进行交换。这对于某些需要特定数据布局的计算非常有用,例如在机器学习和深度学习中调整数据的形状以适应算法要求。

二、语法和参数

语法:

python 复制代码
numpy.transpose(a, axes=None)

参数:

  • a: 输入的数组。
  • axes: 一个包含轴顺序的元组,用于指定新的轴顺序。如果省略此参数,数组将被转置(即轴0和轴1互换,依此类推)。

返回值:

返回一个新的数组视图,其轴顺序已经被置换。

三、实例

3.1 基本转置
  • 代码:
python 复制代码
import numpy as np

# 创建一个 2x3 的数组
arr = np.array([[1, 2, 3], [4, 5, 6]])

# 转置数组
transposed_arr = np.transpose(arr)
  • 输出:
python 复制代码
[[1 4]
 [2 5]
 [3 6]]
3.2 指定轴顺序
  • 代码:
python 复制代码
import numpy as np

# 创建一个 2x3x4 的三维数组
arr_3d = np.array([[[ 0,  1,  2,  3],
                    [ 4,  5,  6,  7],
                    [ 8,  9, 10, 11]],

                   [[12, 13, 14, 15],
                    [16, 17, 18, 19],
                    [20, 21, 22, 23]]])

# 指定新的轴顺序,例如 (2, 1, 0)
transposed_arr_3d = np.transpose(arr_3d, (2, 1, 0))
  • 输出:
python 复制代码
[[[ 0  4  8]
  [ 1  5  9]
  [ 2  6 10]
  [ 3  7 11]]

 [[12 16 20]
  [13 17 21]
  [14 18 22]
  [15 19 23]]]

四、注意事项

  1. numpy.transpose 不会改变原始数组的数据,而是返回一个新的数组视图。
  2. 如果 axes 参数省略,对于二维数组,它将执行常规的矩阵转置操作。
  3. 对于更高维度的数组,axes 参数允许更复杂的轴置换操作。
  4. 当指定 axes 参数时,需要确保提供的轴顺序与原始数组的维度匹配。
  5. 转置操作在内存中是高效的,因为它不涉及数据的实际移动,只是改变了数据的解释方式。
相关推荐
秃头佛爷2 小时前
Python学习大纲总结及注意事项
开发语言·python·学习
dayouziei4 小时前
java的类加载机制的学习
java·学习
dsywws7 小时前
Linux学习笔记之vim入门
linux·笔记·学习
西柚小萌新8 小时前
七.numpy模块
numpy
晨曦_子画8 小时前
3种最难学习和最容易学习的 3 种编程语言
学习
城南vision8 小时前
Docker学习—Docker核心概念总结
java·学习·docker
ctrey_9 小时前
2024-11-1 学习人工智能的Day20 openCV(2)
人工智能·opencv·学习
十年之少9 小时前
由中文乱码引来的一系列学习——Qt
学习
u01015265810 小时前
STM32F103C8T6学习笔记2--LED流水灯与蜂鸣器
笔记·stm32·学习
王俊山IT10 小时前
C++学习笔记----10、模块、头文件及各种主题(二)---- 预处理指令
开发语言·c++·笔记·学习