Question

我有以下有效的代码：

for model in models:
    model.fit(trainset)
    predictions = model.test(testset)
    top_n = get_top_n(predictions, n=5)

    # Print the first one
    user = list(top_n.keys())[0]
    print(f'model: {model}, {user}: {top_n[user]}')
print('Top N computation successful!')

但是挑战是这样-我正在尝试使用该模型预测足球比赛。感兴趣的两列是：home_game和away_game。

这些列中的两个感兴趣的项目是：Chelsea和West Ham。

使用上述模型有条件地选择这两项的最佳方法是什么？

Answer 1

我假设您的数据框看起来像这样：

df = pd.DataFrame({'home_game':['Man Utd', 'Chelsea', 'Arsenal'], 'away_game':['Liverpool','West Ham','Spurs']})

    away_game   home_game
0   Liverpool   Man Utd
1   West Ham    Chelsea
2   Spurs       Arsenal

您可以通过以下方式选择切尔西和西汉姆的行：

test = df[(df['home_game'] == 'Chelsea') & (df['away_game'] == 'West Ham')]

并通过传递结果数据帧进行预测（取决于模型的预测功能）：

model.predict(test)

如何使用预测模型python预测列中的特定行？

1 个答案: