python - 使用 enumerate 计算一个值在 np 数组中的位置

Question

我正在尝试编写一个返回分布 P 中位数的函数：

我的想法是使用 np.median() 然后枚举以获取顺序或索引（在本例中为 x）。有什么建议么？

谢谢！

标签： pythonnumpyprobability

Answer 1

感谢 Paul，我正在寻找的代码是：

def median_cal(P):
    npa = np.asarray(P, dtype=np.float32)
    a = npa.cumsum().searchsorted(0.5)+1
    return a