熊猫数据框过滤出小尺寸的组

Question

我正在尝试过滤掉小尺寸的组，并且过滤功能抛出ValueError：不允许使用负尺寸。 我不能发布我的代码，因为它太具体了，我不能仅凭随机数据重现该问题。 有人遇到过这个吗？ 我看过其他帖子也遇到了同样的错误，但是并没有太大帮助

作为一种变通办法，我正在尝试对filter和groupby进行相同的操作并应用，但是它没有按预期工作。 有什么建议么？

       dff = pd.DataFrame({'A': np.arange(8), 'B': list('aabbbbcc')})
       dff['C'] = np.arange(8)

       def f(x):
         if len(x)>2: 
           return x
         else:
          return None
       dff.groupby('B').apply(f)

       B
        a         A    B    C
        2  NaN  NaN  NaN
        3  NaN  NaN ...
        b       A  B  C
        2  2  b  2
        3  3  b  3
        4  4  b  4
        5 ...
        c         A    B    C
        2  NaN  NaN  NaN
        3  NaN  NaN ...
       dtype: object

Answer 1

您的示例代码的问题在于，当将数据帧放回原处时， apply不知道如何处理None 。 您的apply函数需要每次输出相同类型的对象。 如果返回pd.DataFrame()而不是None，则应该得到所需的内容。