今天为大家带来的内容是:实列代码讲解:解决Numpy中sum函数求和结果维度的问题(必看篇)
使用Numpy(下面简称np)中的sum函数对某一维度求和时,由于该维度会在求和后变成一个数,所以所得结果的这一维度为空。
比如下面的例子:
![5e782fd78bd4f4d04fad9fa0ad2f942d.png](https://i-blog.csdnimg.cn/blog_migrate/a1220aa88d66aa8f61c27f07373eda9d.jpeg)
所以,对于一个shape为(2,3)的数组,在默认情况下使用np.sum函数求和后得到的结果shape是 (2,),如果我们想得到的是(2,1)的shape怎么办?比如Ng的深度学习编程练习中Course 1 Assignment 4就要求这样。使用reshape函数当然可以,只是没有必要,太麻烦了一点不优雅。我们可以使用通过设置keepdims参数实现,还是这个例子:
![e627ce6d934234576774cd267e484fa9.png](https://i-blog.csdnimg.cn/blog_migrate/3d8a94d467413c4b81ad3515b5f65037.jpeg)
嗨喽:正在学习python的小伙伴或者打算学习的,可以私信小编“07”领取资料!
(2,1)和(2,)的shape之间不同参见 What's the difference between (N,) and (N,1) in Numpy? —Stackoverflow
这里有个小例子可以帮助理解:
![3607ef52f2051b0e7acf7bfa8d7f2904.png](https://i-blog.csdnimg.cn/blog_migrate/b206090adf499c98f8e172df1d8c7e88.jpeg)
以上这篇解决Numpy中sum函数求和结果维度的问题就是小编分享给大家的全部内容了,希望能给大家一个参考!!!
最后多说一句,小编是一名python开发工程师,这里有我自己整理了一套最新的python系统学习教程,包括从基础的python脚本到web开发、爬虫、数据分析、数据可视化、机器学习等。想要这些资料的可以关注小编,并在后台私信小编:“07”即可领取。