Pythonsum() 函数

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])htl666 htl666

190***2891@qq.com

6年前 (2019-08-28)