GonnaHetzMe/blitz-tools
0
0 Fork 0
mirror of https://github.com/Jylpah/blitz-tools.git synced 2026-10-21 10:17:05 +02:00
Code Issues Projects Releases Wiki Activity

Fix handling of missing team member stats

Browse Source
This commit is contained in:
Jylpah committed 2018-10-08 21:24:46 +08:00
1 parent 610fbfbd5b
commit 9a191471f4
1 file changed
+9 -11
Download Patch File Download Diff File Expand all files Collapse all files
analyzeReplays.py
+9 -11
View File
@@ -361,28 +361,26 @@ def calcTeamStats(result_list: list, player_stats : dict) -> list:
stat_types = player_stats[list(player_stats.keys())[0]].keys() stat_types = player_stats[list(player_stats.keys())[0]].keys()
for result in result_list: for result in result_list:
try: try:
n_allies = len(result['allies']) n_allies = collections.defaultdict(defaultvalueZero)
allies_stats = collections.defaultdict(defaultvalueZero) allies_stats = collections.defaultdict(defaultvalueZero)
for ally in result['allies']: for ally in result['allies']:
for stat in stat_types: for stat in stat_types:
if player_stats[ally][stat] == None: if player_stats[ally][stat] != None:
n_allies -= 1
else:
allies_stats[stat] += player_stats[ally][stat] allies_stats[stat] += player_stats[ally][stat]
n_allies[stat] += 1
for stat in stat_types: for stat in stat_types:
result['allies_' + str(stat)] = allies_stats[stat] / n_allies result['allies_' + str(stat)] = allies_stats[stat] / n_allies[stat]
n_enemies = len(result['enemies']) n_enemies = collections.defaultdict(defaultvalueZero)
enemies_stats = collections.defaultdict(defaultvalueZero) enemies_stats = collections.defaultdict(defaultvalueZero)
for enemy in result['enemies']: for enemy in result['enemies']:
for stat in stat_types: for stat in stat_types:
if player_stats[enemy][stat] == None: if player_stats[enemy][stat] != None:
n_enemies -= 1
else:
enemies_stats[stat] += player_stats[enemy][stat] enemies_stats[stat] += player_stats[enemy][stat]
n_enemies[stat] += 1
for stat in stat_types: for stat in stat_types:
result['enemies_' + str(stat)] = enemies_stats[stat] / n_enemies result['enemies_' + str(stat)] = enemies_stats[stat] / n_enemies[stat]
return_list.append(result) return_list.append(result)
except KeyError as err: except KeyError as err: