pandas - 如果失败，继续对下一个 groupby 组进行 qcut？

Question

我对 groupby 结果应用 qcut，以下是我的问题的简化版本:

a = pd.DataFrame({'A':[1,1,1,1,2,2,2,2],
                  'B': [0,0,0,0,2,3,7,6]})
a.groupby(['A'])['B'].apply(lambda x: pd.qcut(x, 2))

问题是，一些小组提出了 ValueError 因为

raise ValueError('Bin edges must be unique: %s' % repr(bins))
ValueError: Bin edges must be unique: array([ 0.,  0.,  0.])

我想要实现的是，如果 qcut 失败，跳过(或传递 0 作为结果)并继续下一组的 qcut。有什么建议吗？

Answer 1

@JohnE 的回答略有不同，以允许我们保留 apply 语法。

def try_qcut(x,n):
    try:
        return pd.qcut(x,n)
    except ValueError:
        return x*np.nan

a.groupby('A')['B'].apply(lambda x: try_qcut(x, 2))

0         NaN
1         NaN
2         NaN
3         NaN
4    [2, 4.5]
5    [2, 4.5]
6    (4.5, 7]
7    (4.5, 7]
Name: B, dtype: object