python - 目标和目标我想返回结果(赢、平、输)。 Pandas python

Question

尝试根据 GF 和 GA 创建一个新列“Results”。想知道如何通过定义函数和条件语句来做到这一点。以下是我对一行的初步尝试，但无法弄清楚如何将其应用于所有行，非常感谢您的帮助!

<表类="s-表"><头><日>日期 地点对手GFGA<正文>2017-08-19H Crystal 宫102017-08-27H阿森纳402017-09-16H伯恩利112017-10-14H曼联002017-10-28H哈德斯菲尔德30

原始输入:

df = pd.DataFrame({'Date': ['08-19-2017', '08-27-2017', '09-16-2017', '10-14-2017', '10-28-2017'],
                   'Venue': ['H', 'H', 'H', 'H', 'H'],
                   'Opponent': ['Crystal Palace', 'Arsenal FC', 'Burnley FC', 'Manchester United', 'Huddersfield Town'],
                   'GF': [1, 4, 1, 0, 3],
                   'GA': [0, 0, 1, 0, 0]})

我的尝试:

    # for one row (match)
gf = df[0, 3]
ga = df[0, 4]
        
if gf > ga:
    print('W')
elif gf < ga:
    print('L')
else:
    print('D')
    

输出:W

要将其应用于所有行，我将如何根据以下内容制定解决方案:

 ```# create a function
def win_draw_loss():
    
if df['GF'] > df['GA']:
    print('W')
elif mo_rec['GF'].sum() < mo_rec['GA'].sum():
    print('L')
else:
    print('D')

# making the new pandas column
df['results'] =  ```

Answer 1

您可以稍微更正您的 win_draw_loss 函数并将其应用于 DataFrame df 的每一行。下面更正了您的函数的问题是，如果您想将 win_draw_loss 应用于 DataFrame，则它必须将行数据作为参数，因此您的评估也需要针对行数据。

def win_draw_loss(x):
    
    if x['GF'] > x['GA']:
        return 'W'
    elif x['GF'] < x['GA']:
        return 'L'
    else:
        return 'D'

df['Results'] = df.apply(win_draw_loss, axis=1)

你也可以使用np.select:

import numpy as np
df['Results'] = np.select([df['GF'] > df['GA'], df['GF'] < df['GA']], ['W', 'L'], 'D')

输出:

        Date Venue           Opponent  GF  GA  Results
0  08-19-2017     H     Crystal Palace   1   0       W
1  08-27-2017     H         Arsenal FC   4   0       W
2  09-16-2017     H         Burnley FC   1   1       D
3  10-14-2017     H  Manchester United   0   0       D
4  10-28-2017     H  Huddersfield Town   3   0       W