sum是python中一個很實用的函數,但是要注意它的使用,我第一次用的時候,就把它這樣用了:
1
|
s = sum ( 1 , 2 , 3 ) |
結果就悲劇啦
其實sum()的參數是一個list
例如:
1
2
|
sum ([ 1 , 2 , 3 ]) sum ( range ( 1 , 11 )) |
還有一個比較有意思的用法
1
2
3
4
|
a = range ( 1 , 11 ) b = range ( 1 , 10 ) c = sum ([item for item in a if item in b]) print c |
輸出:
1
|
|
現在對於數據的處理更多的還是numpy。沒有axis參數表示全部相加,axis=0表示按列相加,axis=1表示按照行的方向相加
- >>> import numpy as np
- >>> a=np.sum([[0,1,2],[2,1,3]])
- >>> a
- 9
- >>> a.shape
- ()
- >>> a=np.sum([[0,1,2],[2,1,3]],axis=0)
- >>> a
- array([2, 2, 5])
- >>> a.shape
- (3,)
- >>> a=np.sum([[0,1,2],[2,1,3]],axis=1)
- >>> a
- array([3, 6])
- >>> a.shape
- (2,)