import numpy as np a = np.array([[1,2],[3,4]]) # 按行相加,并且保持其二维特性 print(np.sum(a, axis=1, keepdims=True)) # 按行相加,不保持其二维特性 print(np.sum(a, ……
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)