我有一個布爾型別的 NumPy 陣列“boolarr”。我想計算每行中值為 True 的元素的數量。是否有專門用于此任務的 NumPy 或 Python 例程？
例如，考慮下面的代碼：
>>> 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)
每行的計數將給出以下結果：
1
2
2
uj5u.com熱心網友回復：
In [48]: boolarr = np.array([[0, 0, 1], [1, 0, 1], [1, 0, 1]], dtype=bool)
In [49]: boolarr
Out[49]:
array([[False, False, True],
[ True, False, True],
[ True, False, True]])
只需使用sum：
In [50]: np.sum(boolarr, axis=1)
Out[50]: array([1, 2, 2])
加法時的True計數。1
要么：
In [54]: np.count_nonzero(boolarr, axis=1)
Out[54]: array([1, 2, 2])
轉載請註明出處，本文鏈接：https://www.uj5u.com/qita/448645.html