python - 如何在熊猫中使用条件执行分组和转换计数

Question

我有以下数据框：

# Import pandas library 
import pandas as pd
import numpy as np

# data
data = [['tom', 10,2,'c',100,'x'], ['tom',16 ,3,'a',100,'x'], ['tom', 22,2,'a',100,'x'],
        ['matt', 10,1,'c',100,'x'], ['matt', 15,5,'b',100,'x'], ['matt', 14,1,'b',100,'x']]

# Create the pandas DataFrame 
df = pd.DataFrame(data, columns = ['Name', 'Attempts','Score','Category','Rating','Other'])
df['AttemptsbyRating'] = df.groupby(by=['Rating'])['Attempts'].transform('count')
df

然后我尝试创建额外的列 - 一个显示按评级分组的尝试计数（如上所示），然后尝试做另一个我想计算大于 1 的分数。我试过：

df['scoregreaterthan1'] = df[df.groupby(by=['Rating'])['Score'].transform('count')>1]

我得到了一个ValueError: Wrong number of items passed 7, placement implies 1

基本上在上表中，我希望它每列显示 4 个（4 个分数大于 1）

任何帮助将非常感激！谢谢

标签： pythonpandas

Answer 1

我们应该做

df['scoregreaterthan1'] = df['Score'].gt(1).groupby(df['Rating']).transform('sum')