python – numpy的bincount函数的逆

发布时间：2020-12-20 12:31:02 所属栏目：Python 来源：网络整理

导读：给定一个整数计数c的数组,如何将其转换为整数inds数组,使得np.all(np.bincount(inds)== c)为真？例如： c = np.array([1,3,2,2]) inverse_bincount(c) # -- what I needarray([0,1,3]) 上下文：我正在尝试跟踪多组数据的位置,同时对所有数据进行计算.我将所

给定一个整数计数c的数组,如何将其转换为整数inds数组,使得np.all(np.bincount(inds)== c)为真？

例如：

>>> c = np.array([1,3,2,2])
>>> inverse_bincount(c)  # <-- what I need

array([0,1,3])

上下文：我正在尝试跟踪多组数据的位置,同时对所有数据进行计算.我将所有数据连接在一起进行批处理,但我需要一个索引数组来提取结果.

目前的解决方法：

def inverse_bincount(c):
  return np.array(list(chain.from_iterable([i]*n for i,n in enumerate(c))))

解决方法

使用 numpy.repeat：

np.repeat(np.arange(c.size),c)

（编辑：李大同）

【声明】本站内容均来自网络，其相关言论仅代表作者个人观点，不代表本站立场。若无意侵犯到您的权利，请及时与联系站长删除相关内容!