-
Notifications
You must be signed in to change notification settings - Fork 0
/
champions.py
64 lines (52 loc) · 1.68 KB
/
champions.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
N = 140
import json
FILE = open("static_data/champions.json")
championJSON = json.load(FILE)
FILE.close()
championJSON = championJSON["data"]
championTable = {}
FILE = open("static_data/championFeatures.json", "r")
championFeatures = json.load(FILE)
FILE.close()
for name in championJSON.keys():
key = int(championJSON[name]['id'])
championTable[key] = name
def id_to_name(x: int) -> str:
return championTable[x]
# coverts champion name to 8 bit representation
def name_to_index(name: str) -> list:
return championFeatures[name]
# this is pretty inefficient, but whatever
def index_to_name(x: int) -> str:
for name in championFeatures.keys():
if x == championFeatures[name]:
return name
# given a list of names, and converts to features
def team_to_features(team: list) -> list:
returnTeam = [0] * N
for name in team:
index = name_to_index(name)
returnTeam[index] = 1
return returnTeam
# binaries = []
# for name in team:
# binaries.append(championFeatures[name])
# binaries.sort(key=lambda x: x[0])
# binaries = list(map(lambda x: x[1:], binaries))
# for i in range(4):
# binaries[0].extend(binaries[i])
# return binaries[0]
# code to create initial conversion
if __name__ == "__main__":
print("not running main because commented")
# dictionary = {}
# counter = 0
# names = championJSON.keys()
# names = list(names)
# names.sort()
# for name in names:
# dictionary[name] = counter
# counter += 1
# FILE = open("static_data/championFeatures.json", 'w')
# FILE.write(json.dumps(dictionary))
# FILE.close()