Cách đếm số phần tử thực trong mảng bool NumPy

Tôi có một mảng NumPy 'boolarr' thuộc loại boolean. Tôi muốn đếm số phần tử có giá trị là True. Có một thói quen NumPy hoặc Python dành riêng cho nhiệm vụ này không? Hoặc, tôi có cần lặp lại các yếu tố trong kịch bản của mình không?

Bạn có nhiều lựa chọn. Hai lựa chọn như sau.

numpy.sum(boolarr)
numpy.count_nonzero(boolarr)

Đây là một ví dụ:

>>> import numpy as np
>>> boolarr = np.array([[0, 0, 1], [1, 0, 1], [1, 0, 1]], dtype=np.bool)
>>> boolarr
array([[False, False,  True],
       [ True, False,  True],
       [ True, False,  True]], dtype=bool)

>>> np.sum(boolarr)
5

Tất nhiên, đó là một boolcâu trả lời cụ thể. Tổng quát hơn, bạn có thể sử dụng numpy.count_nonzero.

>>> np.count_nonzero(boolarr)
5

Câu hỏi đó đã giải quyết một câu hỏi khá giống tôi và tôi nghĩ mình nên chia sẻ:

Trong python thô, bạn có thể sử dụng sum()để đếm Truecác giá trị trong list:

>>> sum([True,True,True,False,False])
3

Nhưng điều này sẽ không hoạt động:

>>> sum([[False, False, True], [True, False, True]])
TypeError...

Về mặt so sánh hai mảng numpy và đếm số lượng trận đấu (ví dụ: dự đoán lớp chính xác trong học máy), tôi thấy ví dụ dưới đây cho hai chiều hữu ích:

import numpy as np
result = np.random.randint(3,size=(5,2)) # 5x2 random integer array
target = np.random.randint(3,size=(5,2)) # 5x2 random integer array

res = np.equal(result,target)
print result
print target
print np.sum(res[:,0])
print np.sum(res[:,1])

có thể được mở rộng đến kích thước D.

Kết quả là:

Sự dự đoán:

[[1 2]
 [2 0]
 [2 0]
 [1 2]
 [1 2]]

Mục tiêu:

[[0 1]
 [1 0]
 [2 0]
 [0 0]
 [2 1]]

Đếm dự đoán đúng cho D = 1: 1

Đếm dự đoán đúng cho D = 2: 2