python - np.any(np.not_equal(arr, 0), axis=0) 与 np.count_nonzero(arr)

Question

我有一个包含很多非零值的数组。当我使用以下方法计算非零项目总数时，我得到了 2 个不同的结果:

1)

non_zero_weights = np.any(np.not_equal(lr_l1.coef_, 0), axis=0)
new_vocab_size = non_zero_weights.sum()

2)

np.count_nonzero(lr_l1.coef_)

我在这里错过了什么吗？

Answer 1

这是正确的:

np.count_nonzero(lr_l1.coef_)

但是，在第一个选项中，您不是查找数组中非零值的数量，而是查找是否存在至少 1 个非零值。这就是 np.any() 的作用。

这应该是以下内容:

non_zero_weights = np.not_equal(lr_l1.coef_, 0)
new_vocab_size = non_zero_weights.sum()