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

Question

我正在尝试编写一个返回分布 P 的中位数的函数：我的想法是使用 np.median() 然后枚举来获取顺序或索引（在本例中为 x）。有什么建议么？

Answer 1

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

def median_cal(P):

    npa = np.asarray(P, dtype=np.float32)

    a = npa.cumsum().searchsorted(0.5)+1

    return a