Python numpy 中 keepdims 的含義

keepdims主要用於保持矩陣的二維特性

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

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

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

 

輸出

array([[3], [7]])
array([3, 7])

--------------------------------------------------------------

轉載自：http://blog.csdn.net/u012560212/article/details/78393836