Python如何调整数组的形状

news/2024/12/21 22:16:17/

文章目录

    • 更改维度
    • 调整坐标轴
    • 牛刀小试
    • Numpy函数

调整形状
调整形状reshape, resize, flatten, ravel, squeeze
调整坐标轴transpose, swapaxes

更改维度

数组中的数据在内存里是固定的,但计算时的排列方式却可以随时更改,这也是数组的强大之处。其中,reshaperesize功能相同,区别是前者返回新数组,后者则直接修改原始数组。

>>> x = np.arange(12)
>>> y = x.reshape(2,6)
>>> print(x)
[ 0  1  2  3  4  5  6  7  8  9 10 11]
>>> print(y)
[[ 0  1  2  3  4  5][ 6  7  8  9 10 11]]
>>> x.resize(2,6)
>>> print(x)
[[ 0  1  2  3  4  5][ 6  7  8  9 10 11]]

-1表示自动规划某一轴的尺寸,例如

>>> x.reshape(3,-1)
array([[ 0,  1,  2,  3],[ 4,  5,  6,  7],[ 8,  9, 10, 11]])

flattenravel相当于reshape(-1),即将数组展平为一维数组。

squeeze则比flatten稍微温和一点,会删除尺寸为1的维度,例如

>>> x.resize(1,3,4,1,1)
>>> print(x)
[[[[[ 0]][[ 1]][[ 2]][[ 3]]][[[ 4]][[ 5]][[ 6]][[ 7]]][[[ 8]][[ 9]][[10]][[11]]]]]

上面的这个x有太多层括号,看上去毫无卵用,这个时候可以用squeeze

>>> x.squeeze()
array([[ 0,  1,  2,  3],[ 4,  5,  6,  7],[ 8,  9, 10, 11]])

有木有瞬间清爽了许多。

调整坐标轴

transposeswapaxes用于调整坐标轴,如果用矩阵的视角去理解,那么大致相当于转置。

>>> x
array([[ 0,  1,  2,  3,  4,  5],[ 6,  7,  8,  9, 10, 11]])
>>> x.T
array([[ 0,  6],[ 1,  7],[ 2,  8],[ 3,  9],[ 4, 10],[ 5, 11]])
>>> x.transpose(1,0)
array([[ 0,  6],[ 1,  7],[ 2,  8],[ 3,  9],[ 4, 10],[ 5, 11]])

其中,transpose(1,0)表示将第一个坐标轴和第0个坐标轴交换位置。

牛刀小试

熟练掌握数组形状的变换方法,也就相当于熟悉了张量的运算法则,这对于数据科学来说是非常重要的基础技能。

例如,现有300张图像200x100的图像,想要得到每张图像的列质心。传统思路肯定是跑循环,但众所周知Python的循环效率比较慢,所以最佳方法是300张一起做,无非就是300x200x100的张量,对第二个坐标轴进行质心提取而已

imgs = np.random.rand(300,200,100)
xs = np.arange(100)
xCen = np.matmul(imgs, xs) / np.sum(imgs, axis=2)

其中,xCen就是所要求的质心。

当然,也可以用更加直观的做法

xCen = imgs.reshape(-1,100)@xs / np.sum(imgs.reshape(-1,100), axis=1)
xCen = xCen.reshape(300,200)

Numpy函数

对于上面这几种数组的内置方法,有一些可直接从numpy中调用,这样的好处是可以直接对非数组格式的数据进行操作,例如

>>> x = list(range(12))
>>> np.reshape(x, (3,4))
array([[ 0,  1,  2,  3],[ 4,  5,  6,  7],[ 8,  9, 10, 11]])

其中,x是一个列表,np.reshape会自动将其转化为数组后再行操作。

同样地,flatten也可以完成数组展平的任务

>>> x = [[i, i+1] for i in range(5)]
>>> x
[[0, 1], [1, 2], [2, 3], [3, 4], [4, 5]]
>>> np.ravel(x)
array([0, 1, 1, 2, 2, 3, 3, 4, 4, 5])

http://www.ppmy.cn/news/4974.html

相关文章

华为配置动态路由ISIS协议

华为配置动态路由ISIS协议一、路由基础知识二、路由器配置接口IP地址(一)配置R1、R2、R3网络(二)配置R1、R2、R3环回网络接口(三)测试直连网络三、启动进程号,配置实体名称(一&#…

设计模式之单例模式

设计模式之单例模式一、单例模式是什么?使用场景二、实现懒汉式饿汉式双检锁/双重校验锁总结一、单例模式是什么? 这种类型的设计模式属于创建型模式,它提供了一种创建对象的最佳方式。 这种模式涉及到一个单一的类,该类负责创建…

蓝桥杯有必要参赛吗?

昨天和群里的小伙伴在群里聊,有的小伙伴竟然说蓝桥杯一等奖没有含量,我也是醉了! 就像去年看了一个号主写的:研究生遍地都是! 放眼全国14亿人口,别说研究生了,本科生占比有多少? “蓝桥杯是我人生中得到…

Fully Convolutional Adaptation Networks for Semantic Segmentation

参考 论文解析之《Fully Convolutional Adaptation Networks for Semantic Segmentation》 - 云社区 - 腾讯云 论文网址:Fully Convolutional Adaptation Networks for Semantic Segmentation 摘要 深度神经网络的最新进展令人信服地证明了在大数据集上学习视觉模…

44_外部SRAM实验

目录 IS62WV51216简介 IS62WV51216框图 IS62WV51216读时序 IS62WV51216写时序 FSMC简介 FSMC寄存器介绍 硬件连接图 实验源码 IS62WV51216简介 IS62WV51216ISSi (Integrated Silicon Solution,Inc)公司生产的一颗16位宽512K (512*16,即1M字节)容量的CMOS静态内存(SRAM…

Shell编程

目录 一、实验目的 二、实验软硬件要求 三、实验预习 四、实验内容(实验步骤、测试数据等) 1、编写shell程序,实现用户自定义输入十个整数,计算从第3个到第7个整数的和 2、编写shell程序,实现创建一个以学号命名的新…

【lssvm回归预测】基于灰狼算法优化最小支持向量机GWO-LSSVM数据预测模型含Matlab源码

✅作者简介:热爱科研的Matlab仿真开发者,修心和技术同步精进,matlab项目合作可私信。 🍎个人主页:Matlab科研工作室 🍊个人信条:格物致知。 更多Matlab仿真内容点击👇 智能优化算法 …

MR案例(2):学生排序(单字段排序、多字段排序)

文章目录一、任务目标1. 准备数据二、实行任务1. 创建Maven项目2. 添加相关依赖3. 创建日志属性文件4. 创建学生实体类5. 创建学生映射器类6. 创建学生归并器类7. 创建学生驱动类8. 启动学生驱动器类,查看结果一、任务目标 MR案例:学生排序(…