我有一个幻想足球(足球)数据的字典,其中元组中的第一个值是价格,第二个是本赛季的预期点。它的一部分可以在下面看到:
'Romeu': [4.5, 57.0],
'Neves': [5.5, 96.0],
'Townsend': [6.0, 141.0],
'Lucas Moura': [7.5, 105.0],
'Martial': [7.5, 114.0],
'David Silva': [7.5, 177.0],
'Fraser': [7.5, 180.0],
'Richarlison': [8.0, 138.0],
'Bernardo Silva': [8.0, 174.0],
'Sigurdsson': [8.0, 187.0],
我想做的是编写一个程序,允许我设置价格限制并返回固定长度的组合,例如n = 5,拥有最高分。
因此,如果我将价格限制设置为32,并且希望有5位玩家,则返回Romeu,Neves,Townsend,Sigurdsson和Fraser。
有人可以给我提示正确的方向吗?我不知道该如何开始。
最佳答案
这是一种蛮力方法,我从115名(我的笔记本电脑上为1分钟42秒)中选择了5名球员进行尝试。将选择人数增加到100名球员中的20名,将需要100,000年以上的时间执行。 50中的20即使需要4天。
from itertools import combinations
# Set the following parameters as desired
nplayers = 5
price = 32
players = {
'Romeu': [4.5, 57.0],
'Neves': [5.5, 96.0],
'Townsend': [6.0, 141.0],
'Lucas Moura': [7.5, 105.0],
'Martial': [7.5, 114.0],
'David Silva': [7.5, 177.0],
'Fraser': [7.5, 180.0],
'Richarlison': [8.0, 138.0],
'Bernardo Silva': [8.0, 174.0],
'Sigurdsson': [8.0, 187.0],
}
if len(players) < nplayers:
raise IndexError("You selected {nplayers} players but there are only {len(players)} to choose from")
# Create a list of all combinations of players, store as triples (name, cost, score)
combos = combinations(((h, *t) for h, t in players.items()), nplayers)
top_score = 0
for c in combos:
if sum(p[1] for p in c) <= price:
score = sum(p[2] for p in c)
if score > top_score:
top_teams = [c]
continue
elif score == top_score:
top_teams.append(c)
if top_score:
print(top_teams)
else:
print(f"You can't afford a team for only {price}")
输出量
[(('Romeu', 4.5, 57.0), ('Neves', 5.5, 96.0), ('Townsend', 6.0, 141.0), ('Fraser', 7.5, 180.0), ('Sigurdsson', 8.0, 187.0))]
关于python - 与Python dict的组合和,我们在Stack Overflow上找到一个类似的问题:https://stackoverflow.com/questions/57469498/