從句子中提取數字並計算平均值。

Question

我有以下數據的片材：

team1,team2,outcome 
AA,BB,BB won by 90 runs 
AA,CC,AA won by 19 runs (D/L method) 
CC,BB,CC won by 26 runs (D/L method) 
AA,BB,BB won by 56 runs 
CC,BB,CC won by 18 runs 

我需要選擇的數值，並計算它們的平均通過TEAM1分組，TEAM2。

這是到現在爲止。很多垃圾數據，因此我只篩選貧困記錄！

df[df['outcome'].str.contains('runs',na=False)].head() 

我想要的結果：

team1 , team2 , AVG(NUMERIC COLUMN FROM 'OUTCOME') 

請建議！

Answer 1

您可以使用extract與鑄造int第一，然後groupby和聚集mean：

df.outcome = df.outcome.str.extract('(\d+)', expand=False).astype(int) 
print (df.groupby(['team1','team2'], as_index=False)['outcome'].mean()) 
    team1 team2 outcome 
0 AA BB  73 
1 AA CC  19 
2 CC BB  22 

類似的解決方案：

s = df.outcome.str.extract('(\d+)', expand=False).astype(int) 
print (s.groupby([df['team1'],df['team2']]).mean().reset_index()) 
    team1 team2 outcome 
0 AA BB  73 
1 AA CC  19 
2 CC BB  22