Binning of data along one axis in numpy
You could use np.apply_along_axis
:
x = np.array([range(20), range(1, 21), range(2, 22)])nbins = 2>>> np.apply_along_axis(lambda a: np.histogram(a, bins=nbins)[0], 1, x)array([[10, 10], [10, 10], [10, 10]])
The main advantage (if any) is that it's slightly shorter, but I wouldn't expect much of a performance gain. It's possibly marginally more efficient in the assembly of the per-row results.