python sum函数numpy_解决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博客。