On 21 October 2017 at 22:32, Eric Wieser <wieser.eric+numpy@gmail.com>
wrote:
David, that doesn’t work, because np.cumsum(mask)[mask] is always equal
to np.arange(mask.sum()) + 1. Robert’s answer is correct.
Of course, you are right. It makes sense in my head now.
