基于numpy中的expand_dims函数用法

yipeiwu_com7年前Python基础

常见的一种应用场景:

条件:假设A的shape为[4, 2],B的shape为[5, 2]

目的:实现A中的每一行, 减去B中的所有行(broadcast操作)。

实现:

A1 = np.expand_dims(A, -2) => A1的shape变为[4, 1, 2]

B1 = np.expand_dims(B, 0) => B1的shape变为[1, 5, 2]

A1 - B1

其他示例:

wh = np.random.randint(1,3, size=(4,2))

np.expand_dims(wh, -2).shape

np.expand_dims(wh, 1).shape

在倒数第2个轴后面(在正数第1个轴后面)插入一个新轴。

以上这篇基于numpy中的expand_dims函数用法就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持【听图阁-专注于Python设计】。

相关文章

15行Python代码实现网易云热门歌单实例教程

15行Python代码实现网易云热门歌单实例教程

0. 引言 马上314情人节就要来了,是否需要一首歌来抚慰你,受伤或躁动的心灵。来吧,今天教你用15行代码搞定热门歌单。学起来并听起来吧。 本文使用的是Selenium模块,它是一个自...

Python外星人入侵游戏编程完整版

Python外星人入侵游戏编程完整版

PYTHON游戏编程外星人入侵的完整实现思路,具体内容如下 准备工作:下载python,比如Anaconda3(64 bit),导入pygame游戏包 1.外星人设置,alie...

Python 带有参数的装饰器实例代码详解

demo.py(装饰器,带参数的装饰器): def set_level(level_num): def set_func(func): def call_func(*ar...

Python numpy中矩阵的基本用法汇总

Python numpy中矩阵的基本用法汇总

Python矩阵的基本用法 mat()函数将目标数据的类型转化成矩阵(matrix) 1,mat()函数和array()函数的区别 Numpy函数库中存在两种不同的数据类型(矩阵ma...

Python json模块使用实例

实际上JSON就是Python字典的字符串表示,但是字典作为一个复杂对象是无法直接传递,所以需要将其转换成字符串形式.转换的过程也是一种序列化过程. 用json.dumps序列化为jso...