【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])
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章