如何計算給定數組列中非零值的數量？

Question

我有一個如下數組：

[[0, 0, 0, 1],
 [1, 0, 0, 0],
 [0, 1, 0, 0],
 [0, 0, 1, 0],
 [0, 0, 0, 1],
 etc

我想確定每行中索引[3]的總和。 例如，在這里我想得到2作為結果。

我正在努力

np.sum(np.count_nonzero(array[:][3], axis=1))

但是我遇到了界外錯誤。 有任何想法嗎？

Answer 1

您需要將數組索引為a[:, 3] （所有行的第三列），然后可以執行以下操作：

# if the array contains only 0 and 1
a[:,3].sum()
# 2

# if the array can have other values besides 0 and 1
np.count_nonzero(a[:,3])
# 2

這是有關numpy索引的更多信息。

Answer 2

只需將列表獲取為n[:,3]正常求和，因為求和為零將不執行任何操作：

>>> numpy.sum(n[:,3])
2