當前位置: 首頁>>編程示例 >>用法及示例精選 >>正文


Python NumPy count_nonzero方法用法及代碼示例

Numpy 的 count_nonzero(~) 方法計算數組中沿給定軸的非零數。

參數

1. a | array-like

要對其執行該方法的數組。

2. axis | inttuple<int> | optional

我們計算非零數的軸。默認情況下,axis=None

返回值

intNumpy array 指示輸入數組中沿給定軸的非零數。

例子

基本用法

np.count_nonzero([1,0,2,0,3])



3

逐行計數非零

要計算 2D Numpy 數組中逐行非零的數量,請使用以下命令:

np.count_nonzero([[1,0,1],[0,1,0]], axis=1)



array([2, 1])

我們得到這個結果是因為第一行有 2 個非零,第二行有 1 個非零。

按列計數非零

要計算 2D Numpy 數組中按列非零的數量,請使用以下命令:

np.count_nonzero([[1,0,1],[0,1,0]], axis=0)



array([1, 1, 1])

我們得到這個結果是因為每列都包含 1 個非零值。

相關用法


注:本文由純淨天空篩選整理自Isshin Inada大神的英文原創作品 NumPy | count_nonzero method。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。