【Python】【Numpy】keepdims用法

• keepdims=True
• 保持矩阵维数不变(被减少的那个轴会以维度1保留在结果中)。

# 举例：按行相加
import numpy as np
a = np.array([[1, 2, 3],[4, 5, 6]])

print(np.sum(a, axis=1, keepdims=True))
print(np.sum(a, axis=1))

# 输出
array([[5], [7], [9]])
array([5, 7, 9])