On Mon, 2018-09-17 at 12:37 +0100, Matthew Brett wrote:
> Is there any reason that np.count_nonzero should not take an axis
> argument? As in:
No, sounds like an obvious improvement, but as also with those, someone
has to volunteer to do it...
Coding it will probably mean adding the NpyIter and possibly fast paths
(not sure about the state of count nonzero), but should not be very
On Mon, Sep 17, 2018, at 9:37 PM, Matthew Brett wrote:
> >>> np.better_count_nonzero([[10, 11], [0, 3]], axis=1)
> array([2, 1])
> It would be much more useful if it did...
You might know about this already, but I not too long ago discovered np.apply_along_axis , which is a magical function that makes all functions axis aware. Obviously I would support adding an axis keyword to count_nonzero, but in the meantime this function gets you there quickly!