python和numpy的sum()函数详解

correct+=(predicted==labels).sum()

一、Python自带sum()

1. sum(iterable[, start])：iterable——可迭代对象，如：list、tuple、set等；start——固定相加的参数，默认为0。

sum([0, 1, 2])
-- > 3

sum([1, 2, 3], 10)  #列表计算总和后再加 10
-- > 15

二、numpy中的sum()

1. 没有axis参数表示全部相加；axis=0表示按列相加；axis=1表示按行相加；
2. 二维数组经过sum()运算后，会变成一维数组，即降维；
3. 二维矩阵经过运算后仍然是二维矩阵

a = [[1,2],[2,3],[3,4]]

sum(a)
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
TypeError: unsupported operand type(s) for +: 'int' and 'list'

np.sum(a)
-- > 15

np.sum(a, axis=0)
-- > array([6, 9])

np.sum(a, axis=1)
-- > array([3, 5, 7])