numpy.sum

numpy.sum 함수는 지정된 축을 따라 어레이 요소의 합을 계산합니다.



예제1

import numpy as np

a = np.sum([0.5, 1.5])
b = np.sum([0.5, 0.7, 0.2, 1.5], dtype=np.int32)

print(a)
print(b)

2.0
1

a는 [0.5, 1.5] 어레이 요소의 합 2.0을 반환합니다.

b의 경우 dtype을 np.int32로 설정해서, 0.5, 0.7, 0.2는 0이 되고 마지막 요소 1이 모든 요소의 합이 됩니다.



예제2

import numpy as np

a = np.sum([[0, 1], [0, 5]])
b = np.sum([[0, 1], [0, 5]], axis=0)
c = np.sum([[0, 1], [0, 5]], axis=1)
d = np.sum([])

print(a)
print(b)
print(c)
print(d)

6
[0 6]
[1 5]
0.0

축을 지정하지 않으면 모든 요소의 합을 구합니다.

축을 지정하면, 축을 따라 요소의 합을 구하고, 빈 어레이의 경우 0.0을 반환합니다.



이전글/다음글

이전글 :
다음글 :

공유하기