解决Numpy中sum函数求和结果维度的问题
更新时间：2019年12月06日 10:41:50 作者：紫金港浪里小白条
今天小编大家分享一篇解决Numpy中sum函数求和结果维度的问题，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
使用Numpy（下面简称np）中的sum函数对某一维度求和时，由于该维度会在求和后变成一个数，所以所得结果的这一维度为空。
比如下面的例子：
a = np.array([[1,2,3],[4,5,6]]) b = np.sum(a,axis=1) print(b.shape) # (2,)
所以，对于一个shape为(2,3)的数组，在默认情况下使用np.sum函数求和后得到的结果shape是 (2,)，如果我们想得到的是(2,1)的shape怎么办？比如Ng的深度学习编程练习中Course 1 Assignment 4就要求这样。使用reshape函数当然可以，只是没有必要，太麻烦了一点不优雅。我们可以使用通过设置keepdims参数实现，还是这个例子：
a = np.array([[1,2,3],[4,5,6]]) b = np.sum(a,axis=1,keepdims=True) print(b.shape) # (2,1)
(2,1)和(2,)的shape之间不同参见 What's the difference between (N,) and (N,1) in Numpy? —Stackoverflow
这里有个小例子可以帮助理解：
a = np.ones((5,)) b = np.ones((5,1)) print(a) # [1. 1. 1. 1. 1.] print(b) # [[1.] # [1.] # [1.] # [1.] # [1.]]
以上这篇解决Numpy中sum函数求和结果维度的问题就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。
相关文章
本篇文章给大家详细讲述了python中使用subprocess批量执行linux命令的方法，有兴趣的朋友参考学习下。2018-04-04
这篇文章主要介绍了使用Python编写基于DHT协议的BT资源爬虫的方法,文中对于DHT协议的相关知识也作了补充说明,需要的朋友可以参考下2016-03-03
在本篇内容里小编给大家整理的是关于python交互模式是什么的相关基础知识点，需要的朋友们可以参考下。2020-06-06
这篇文章主要介绍了Python中的Xpath和lxml库的使用详解,XPath即 XML路径语言,它是一门在 XML 文档中查找信息的语言,最初被用来搜寻 XML 文档,同时它也适用于搜索 HTML 文档,因此,在爬虫过程中可以使用 XPath 来提取相应的数据,需要的朋友可以参考下2023-12-12
这篇文章主要为大家详细介绍了python学生信息管理系统，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2018-03-03
这篇文章主要介绍了8种用Python实现线性回归的方法对比详解，说到如何用Python执行线性回归，大部分人会立刻想到用sklearn的linear_model，但事实是，Python至少有8种执行线性回归的方法，sklearn并不是最高效的,需要的朋友可以参考下2019-07-07
本文介绍了如何使用 pytest-html 插件生成测试报告,并提供了详细的操作步骤、配置项和示例代码,具有一定的参考价值,感兴趣的可以了解一下2023-11-11
在本篇文章里小编给大家分享了关于python文件编写好后如何实践的相关内容，需要的朋友们可以参考下。2020-07-07
这篇文章主要介绍了Python3.x+迅雷x 自动下载高分电影的实现方法，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-01-01
本篇文章主要介绍了Python中Django 后台自定义表单控件，其实 django 已经为我们提供了一些可用的表单控件，比如：多选框、单选按钮等，有兴趣的开业了解一下。2017-03-03
最新评论