python - 强制 qcut 分成等概率的组

Question

考虑以下数据:

s = pd.Series([1, 1, 1, 2])
pd.qcut(s, 2, labels=False)

我想要等概率组，即 [0, 0, 1, 1]。相反，qcut 可以引发异常或返回 [0, 0, 0, 0](取决于 duplicates kwarg)。

我通过稍微扰乱数据得到了等概率组，即:

def my_qcut(s, *args, **kwargs):
    disturbed = np.array(s) + np.random.rand(len(s)) / 1000
    return pd.qcut(disturbed, *args, **kwargs)

虽然看起来可行，但我觉得这太hack了。谁能就 split 成等概率组的确定性方法向我提供建议？

Answer 1

尝试:

pd.qcut(s.rank(method='first'), 2, labels=np.arange(2))