你可以简单地使用sum:
>>> import numpy
>>> n = numpy.random.randint(0, 2, size=(3,3))
>>> n
array([[1, 0, 1],
[0, 1, 1],
[1, 1, 1]])
>>> n.sum()
7
Run Code Online (Sandbox Code Playgroud)
由于bools对False/True的整数值为0/1,即使数组中的元素不是0或1,您也可以使用此技巧的变体:
>>> n = numpy.random.randint(0, 3, size=(3,3))
>>> n
array([[2, 2, 0],
[0, 2, 0],
[1, 1, 0]])
>>> n == 1
array([[False, False, False],
[False, False, False],
[ True, True, False]], dtype=bool)
>>> (n == 1).sum()
2
Run Code Online (Sandbox Code Playgroud)