Grouping indices of unique elements in numpy

前端 未结 5 627
伪装坚强ぢ
伪装坚强ぢ 2021-01-17 17:15

I have many large (>100,000,000) lists of integers that contain many duplicates. I want to get the indices where each of the element occur. Currently I am doing something li

5条回答
  •  南方客
    南方客 (楼主)
    2021-01-17 17:54

    def to_components(index):
        return np.split(np.argsort(index), np.cumsum(np.unique(index, return_counts=True)[1]))
    

提交回复
热议问题