Python Pandas DataFrame组均值按条件过滤

Question

I'd like to apply the command below to a data frame where the number of groups meet a minimum count criteria.

db=table.groupby(['Type','Quarter'])["Price"].mean()

So far the below example isn't returning the needed results.

db=table.groupby(['Type','Quarter']).filter(lambda group: group.size > 3).groupby(['Type','Quarter'])["SALE_PRC"].mean()

Basically I'd like to find the mean of the["Price"] for the (['Type','Quarter']) groups but only if the number of records exceeds 3.

Appreciate any help. Thank you

Answer 1

You need len(group) for size of groups :

db=table.groupby(['Type','Quarter'])
        .filter(lambda group: len(group) > 3)
        .groupby(['Type','Quarter'])["Price"]
        .mean()

Or use transform :

db=table[table.groupby(['Type','Quarter'])['Type'].transform('size') > 3]
           .groupby(['Type','Quarter'])["Price"].mean()