Python numpy 中 keepdims 的含义

keepdims主要用于保持矩阵的二维特性

 例子

import numpy as np
a = np.array([[0,1],[2,3]])

# 按行相加，并且保持其二维特性
print(np.sum(a, axis=1, keepdims=True))

# 按行相加，不保持其二维特性
print(np.sum(a, axis=1))

结果

array([[1], [5]])
array([1, 5])