python - 将 N 个点放入 M 个相等的箱子中

Question

我有一个包含 N 个(正)点的数组。我想找到直方图的 M 个 bin 边缘，使所有条形具有相同的高度。换句话说，我想找到 M+1 个点，使得两个连续 bin 边缘之间的数组点数相同。

示例

>>> array = [0.3 0.3 0.3 0.7 0.8 0.9]
>>> M = 2
>>> binPartition(array, M)
[0, 0.5, 1]

我会很感激 python 和 numpy 的答案，但一个已知算法的链接就足够了!谢谢你! :)

Answer 1

用 percentile 做不到:

import numpy as np

def binPartition(array, M):
    return np.percentile(array, np.linspace(0, 100, M + 1))

binPartition([0.3, 0.3, 0.3, 0.7, 0.8, 0.9], 2)
>>> array([ 0.3,  0.5,  0.9])