【NumPy中数组广播的拓展】


这是本人在np学习中自定义函数时遇到的一些问题,然后实现的一个小功能

一、 np中广播问题的实现规则

你好! 这是你第一次使用 Markdown编辑器 所展示的欢迎页。如果你想学习如何使用Markdown编辑器, 可以仔细阅读这篇文章,了解一下Markdown的基本语法知识。

1、两个数组维度相同,且大小相同时

这个很容易理解

2、两个数组各维度大小从后往前比对均一致(这里要注意,一定是从后往前)

如:
A = np.zeros((5,1,7))
B = np.zeros((1,7))
print((A+B).shape)
# 输出 (5, 1, 7)

3、两个数组存在一些维度大小不相等时,有一个数组的该不相等维度大小为1

如:
A = np.zeros((1,8,7,4))
B = np.zeros((7,1))
print((A+B).shape)
# 输出:(1, 8, 7, 4)
可以看到,此时仍然是从后往前去对比的。

二、np中广播不能实现的情况

两个数组维度相同,但大小不同时

如:
A = np.zeros((5,2))
B = np.zeros((3,4))
print((A+B).shape)
# 输出:
ValueError: operands could not be broadcast together with shapes (5,2) (3,4)
A = np.zeros((5,2,3))
B = np.zeros((3,4,7))
print((A+B).shape)
# 输出:
ValueError: operands could not be broadcast together with shapes (5,2,3) (3,4,7))

三、如何实现这种维度相同,但大小不同且不是1维的情况呢?

思路:	应该将两个数组中行数较小的,广播成为行数较大的数组的行数。
    	将两个数组中列数较小的,广播成为列数较大的数组的列数
     	当两个数组的维度一样时,就可以直接运算了

这里面的+号可以换成其他的操作符
此时:
A = np.zeros((5,2,3))
B = np.zeros((3,4,7))
print(Add(A,B).shape)
# 输出:
(5, 4, 7)
这样就可以拓展numpy中无法实现的同维度,但是不同大小的数组的操作了。
对于其他维度仍然适用,但是由于精力有限,目前没有解决不同维度,且大小不同的两个数组相加的情况。

  • 3
    点赞
  • 9
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值