Increment Numpy array with repeated indices

前端 未结 4 660
渐次进展
渐次进展 2020-11-28 13:52

I have a Numpy array and a list of indices whose values I would like to increment by one. This list may contain repeated indices, and I would like the increment to scale wit

4条回答
  •  爱一瞬间的悲伤
    2020-11-28 14:09

    If b is a small subrange of a, one can refine Alok's answer like this:

    import numpy as np
    a = np.zeros( 100000, int )
    b = np.array( [99999, 99997, 99999] )
    
    blo, bhi = b.min(), b.max()
    bbins = np.bincount( b - blo )
    a[blo:bhi+1] += bbins
    
    print a[blo:bhi+1]  # 1 0 2
    

提交回复
热议问题