sum函数中keepdims参数的目的是为保持输出结果的维度,然而对于一个矩阵来说,它总是二维的,所以若输入的对象类为矩阵,那么输出就一定是一个矩阵,而矩阵一定是二维的,所以keepdims参数在此时是没有意义的。 这就是它报错的原因。
具体情况如下:
a = np.matrix([[1, 2, 3], # 此时a为一个矩阵
[4, 5, 6],
[7, 8, 9]])
np.sum(a, axis = 0)
# 输出矩阵
matrix([[12, 15, 18]])
np.sum(a, axis = 0 ,keepdims = True)
# 报错TypeError: sum() got an unexpected keyword argument 'keepdims'
# 报错原因是输出已经为矩阵,所以keepdims参数无意义
所以若输入的对象类为二维数组,那么输出就是一个一维数组,此时维度发生了改变,所以keepdims参数在此时就发挥了作用。
具体情况如下:
b = np.array([[1, 2, 3], # 此时b为一个二维数组
[4, 5, 6],
[7, 8, 9]])
np.sum(b, axis = 0)
# 输出一维数组
array([12, 15, 18])
np.sum(b, axis = 0 ,keepdims = True)
# 输出array([[12, 15, 18]])
# 成功将一维变成二维