Pandas：使用groupby按特定值计算比率

Question

你好我有一个看起来像这样的数据框：

我想计算 'count_number' 列中的比率，基于此公式的 'tone' 列中的值：['blue'+'grey']/'red' 每个 'participant_id'、'session'、'block' -

的单位组合

这是我的数据集的一部分作为文本，左列 'RATIO' 是我的预期输出：

participant_id session block tone count_number RATIO 10 1 neg blue 0 0 10 1 neg grey 0 0 10 1 neg red 3 0 10 1 neu blue 1 #DIV/0! 10 1 neu grey 1 #DIV/0! 10 1 neu red 0 #DIV/0! 10 2 neg blue 3 2.333333333 10 2 neg grey 4 2.333333333 10 2 neg red 3 2.333333333 10 2 neu blue 4 1.333333333 10 2 neu grey 0 1.333333333 10 2 neu red 3 1.333333333 11 1 neg blue 0 0 11 1 neg grey 0 0 11 1 neg red 3 0

我试过这个（错误的）方向

def group(df):
  grouped = df.groupby(["participant_id", "session", "block"])['count_number']
  return grouped

neutral = df.loc[df.tone=='grey']
pleasant = df.loc[df.tone=='blue']
unpleasant = df.loc[df.tone=='red']

df['ratio'] = (group(neutral)+group(pleasant)) / group(unpleasant)

Answer 1

这是一种方法：

我们可以为除法的分子和分母创建单独的 Series 对象；然后 groupby + transform sum + div 将获取所需的比率：

num = df['tone'].isin(['blue','grey']) * df['count_number']
denom = df['tone'].eq('red') * df['count_number']
cols = [df[c] for c in ['participant_id', 'session', 'block']]
df['RATIO'] = (num.groupby(cols).transform('sum')
               .div(denom.groupby(cols).transform('sum'))
               .replace(float('inf'), '#DIV/0!'))

另一种方法可能是使用 groupby + 应用计算每个组所需比率的 lambda；然后将比率映射回原始 DataFrame：

cols = ['participant_id', 'session', 'block']
mapping = (df.groupby(cols)
           .apply(lambda x: (x.loc[x['tone'].isin(['blue','grey']), 'count_number'].sum() / 
                             x.loc[x['tone'].eq('red'), 'count_number']))
           .droplevel(-1))

df['RATIO'] = df.set_index(cols).index.map(mapping)
df['RATIO'] = df['RATIO'].replace(float('inf'), '#DIV/0!')

输出：

    group  participant_id  session block  tone  count_number     RATIO
0       1              10        1   neg  blue             0       0.0
1       1              10        1   neg  grey             0       0.0
2       1              10        1   neg   red             3       0.0
3       1              10        1   neu  blue             1   #DIV/0!
4       1              10        1   neu  grey             1   #DIV/0!
5       1              10        1   neu   red             0   #DIV/0!
6       1              10        2   neg  blue             3  2.333333
7       1              10        2   neg  grey             4  2.333333
8       1              10        2   neg   red             3  2.333333
9       1              10        2   neu  blue             4  1.333333
10      1              10        2   neu  grey             0  1.333333
11      1              10        2   neu   red             3  1.333333
12      1              11        1   neg  blue             0       0.0
13      1              11        1   neg  grey             0       0.0
14      1              11        1   neg   red             3       0.0

Pandas：使用groupby按特定值计算比率

Pandas: using groupby to calculate a ratio by specific values

python

aggregate

dataframe

pandas

pandas-groupby