Python-Numpy库的扩维(维度扩展)操作示例代码积累

Posted 昊虹图像算法

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了Python-Numpy库的扩维(维度扩展)操作示例代码积累相关的知识,希望对你有一定的参考价值。

Numpy库中有不少与索引相关的方法或函数要求索引的维度与输入数组的维度一致,比如函数take_along_axis()和函数put_along_axis()都要求这样。

所以本篇博文记录下扩维(维度扩展)的操作示例代码,以便自己和大家需要时方便取用。

第01种方法:使用函数expand_dims()

import numpy as np

A = np.array([[10, 30, 20], [60, 40, 50]])
A_max_index = np.argmax(A, axis=1)

A_max_index_expand = np.expand_dims(A_max_index, axis=1)

运行结果如下:

第02种方法:使用None对象
示例代码如下:

import numpy as np

A = np.array([[10, 30, 20], [60, 40, 50]])
A_max_index = np.argmax(A, axis=1)

A_max_index_expand = A_max_index[:, None]

运行结果如下:

以上是关于Python-Numpy库的扩维(维度扩展)操作示例代码积累的主要内容,如果未能解决你的问题,请参考以下文章

Python-Numpy库的学习

python-Numpy学习之ndimshapedtypeastype的用法

神经网络基本原理简明教程-0-Python-Numpy库的点滴

详解Python-Numpy库的函数diagonal()并附函数diagonal()与函数diag()的区别

ECS集群基于CPU/内存的扩缩容任务

PyTorch:tensor-张量维度操作(拼接维度扩展压缩转置重复……)