numpy - 带索引的 Bincount

Question

我正在寻找一种有效的方法来计算 bincount 的分箱索引作为 ndarray。

为了显示：

>>> x = np.array([0, 1, 1, 0, 2])
>>> b = np.bincount(x)
>>> b
[2 2 1]

我现在正在寻找一个代表每个 bin 元素索引的 ndarray：

[0 3 1 2 4]

我正在寻找一个不应包含循环的快速 numpy 解决方案。任何人都知道如何实现这一点？首先十分感谢！

标签： numpy