python numpy array 数组维度转换（转维）

在实践中，经常需要对中间数据或输出数据进行维度转换，保证多个数据间计算维度上的一致性。

一般主要涉及pytorch中tensorde的维度转换、numpy中array的维度转换。本篇先对array的维度转换方法进行总结，涉及的转换方法如下：

# reshape()
# ravel()
# flatten()
# transpose()
# resize()

具体用法：

array_01 = np.arange(24)
print('\narray_01:\n', array_01, '\narray shape:', array_01.shape)# 输出：array_01是一个一维数组，数组元素24个
# array_01:
# [ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23]
# array shape: (24,) # reshape(n1,n2,...，ni)将数组转换成特定维度
# n1*n2*...*ni = 转换前数组元素的总数
array_02 = array_01.reshape(3, 2, 4)
print('\narray_02:\n', array_02, '\narray shape:', array_02.shape)# 输出：array_02是一个三维数组，channel为3，2行4列，3*2*4=24
# array_02:
# [[[ 0  1  2  3]
#  [ 4  5  6  7]]
#
# [[ 8  9 10 11]
#  [12 13 14 15]]
#
# [[16 17 18 19]
#  [20 21 22 23]]]
# array shape: (3, 2, 4)# ravel将多维数组转换为一维
array_03 = array_02.ravel()
print('\narray_03:\n', array_03, '\narray shape:', array_03.shape)# 输出：array_03是一个一维数组，数组元素24个
# array_03:
# [ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23]
# array shape: (24,)# flatten，与ravel相同，区别在是否拷贝
array_04 = array_02.flatten()
print('\narray_04:\n', array_04, '\narray shape:', array_04.shape)# 输出：array_04是一个一维数组，数组元素24个
# array_04:
# [ 0  1  2  3  4  5  6  7  8  9 10 11 12 13 14 15 16 17 18 19 20 21 22 23]
# array shape: (24,)# transpose转换坐标系,transpose(1, 2, 0)括号里的值表示坐标系的索引号
# 原始的坐标系为[c,h,w]  0-c,1-h,2-w
# 变换后的坐标系统[h,w,c]
array_05 = array_02.transpose(1, 2, 0)
print('\narray_05:\n', array_05, '\narray shape:', array_05.shape)# 输出:array_05是一个三维数组
# array_05:
# [[[ 0  8 16]
#  [ 1  9 17]
#  [ 2 10 18]
#  [ 3 11 19]]
#
# [[ 4 12 20]
#  [ 5 13 21]
#  [ 6 14 22]
#  [ 7 15 23]]]
# array shape: (2, 4, 3)

python numpy array 数组维度转换（转维）相关推荐

python numpy array中维度的区别 array.shape
运行以下代码: 3 import numpy as np4 5 aa = np.array([6 [1,2,3,4],7 [5,6,7,8],8 [9,8,7,6]9 ]) 10 11 r1 = aa ...
Python | Numpy三维数组维度变换/提取
0. 问题描述每次使用Numpy,遇到需要从(A,B,C)三维数组中提取(A,B).(A,C)或者(B,C)或者(A,)这几个维度数据时,总是忘记该如何切片,网上搜到的又太详细,这里做个笔记方便自己 ...
C#数组维度转换：一维数组二维数组三维数组相互转换
最近项目需要用到数组维度转换,网上资源较少,写一个分享与备用 1.一维数组转二维数组 /// <summary> /// 一维数组转2维数组(矩阵) /// </summary> ...
python numpy.array 与list类似，不同点：前者区分元素不用逗号，中间用空格,矩阵用[]代表行向量，两个行向量中间仍无逗号；而list区分元素用逗号
python numpy.array 与list类似,不同点:前者区分元素不用逗号,中间用空格,矩阵用[]代表行向量,两个行向量中间仍无逗号: 而list区分元素用逗号.而 numpy.array 的 ...
python：array和list转换以及数组切片
Python中,数组array和列表list的转换很直接. import numpy as np 1. 首先建立list aaa = [[1,2,3],[4,5,6],[7,8,9],[10,11,1 ...
python numpy array转置_Python numpy数组转置与轴变换
这篇文章主要介绍了Python numpy数组转置与轴变换,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下矩阵的转置 >>> im ...
numpy的array数组的数据升维与降维使用方法自学总结
本文主要介绍numpy中数据升维与降维问题升维注意数组的形式是n行1列还是1行n列 1.numpy.atleast_2d(数组名) 将输入视为至少具有二维的数组 2.numpy.atleast ...
python numpy array索引_Python NumPy 数组索引
1.访问数组元素数组索引与访问数组元素相同. 您可以通过引用其索引号来访问数组元素. NumPy数组中的索引以0开头,这意味着第一个元素的索引为0,第二个元素的索引为1等. 例如: 从以下数组中获取 ...
[转载] python numpy 子数组_Python学习笔记3：Numpy入门
参考链接: Python中的numpy.logaddexp2 参考<Python:数据科学手册>一书,仅作个人学习及记录使用,若有侵权,请联系后台删除. 1 理解Python中的数据类型 ...

python numpy array 数组维度转换（转维）

python numpy array 数组维度转换（转维）相关推荐

最新文章

热门文章