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 sum函数numpy_解决Numpy中sum函数求和结果维度的问题
使用Numpy(下面简称np)中的sum函数对某一维度求和时,由于该维度会在求和后变成一个数,所以所得结果的这一维度为空. 比如下面的例子: a = np.array([[1,2,3],[4,5,6] ...
sum函数python后面很多小数_解决Numpy中sum函数求和结果维度的问题
使用Numpy(下面简称np)中的sum函数对某一维度求和时,由于该维度会在求和后变成一个数,所以所得结果的这一维度为空. 比如下面的例子: a = np.array([[1,2,3],[4,5,6] ...
numpy中sum函数求和时参数axis=0和axis=1的含义
简单来说,对于一个m*n的矩阵,如果sum()的参数中没有指定axis,就是将所有的数值加到一起,得到1*1的标量.如果是axis=0,就是按照列进行加,得到一个1*n的行向量:如果axis=1,表示 ...
Numpy中sum函数的使用方法（Python自带sum函数）
Numpy中sum函数(Python自带sum函数)的作用是对元素求和. 无参时,所有全加: axis=0,按列相加: axis=1,按行相加: 下边通过例子来说明其用法: #!/usr/bin/en ...
matlab中sum函数求和
matlab中sum函数求和 2016_11_4 http://blog.sina.com.cn/charyul 转载 ▼ sum(x,2)表示矩阵x的横向相加,求每行的和,结果是列向量. 而缺省的 ...
mysql 语句中 sum函数求和 null 变 0
在SQL语句中使用集合函数count(),返回结果肯定为非负整数,这也是用来判断数据有无的一个很有效的方法: 然后在用sum()进行集合相加的时候,如果sum()的对象没有记录,那么不会返回0,而返回 ...
python中sum函数详解
python3中sum的用法 python内置的sum函数对象是列表,元组,集合,可以混合起来使用 >>>sum([1,2,3],2)8>>>sum((1,2,3 ...
matlab中sum函数对矩阵的求和总结
matlab中sum函数对矩阵的求和总结 A= [1, 2 ,3 ,4, 5: 1, 2, 3, 4, 5]: a=sum(A) %对整个矩阵按列求和 >>a= [2 4 6 8 ...
Matlab中sum函数使用
今天使用matlab中sum函数,记录一下: sum(A)函数主要功能为针对A的元素求和. 1 如果A为矩阵(matrix): 例如: A=[1 2 3; ...4 9 6;] ; (1)sum(A, ...

Numpy中sum函数求和结果维度问题

Numpy中sum函数求和结果维度问题相关推荐

最新文章

热门文章