wordle_help/aggregate_best.py at main · Suhasr76/wordle_help · GitHub

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

l_100 = [1, 2, 3]
l_1000 = [1, 2, 3]


class Aggregate_best:

    def find_average_dict(self, prefix, l_test):
        average = []
        finalFilename = prefix+"final.txt"

        dict = {}
        for number in l_test:
            filename = prefix+str(number)+'.txt'
            f = open(filename, 'r')
            while True:
                line = f.readline()
                if not line:
                    break
                attributes = line.rstrip().split("\t")
                if attributes[0] in dict:
                    dict[attributes[0]].append(attributes[1:])
                else:
                    dict[attributes[0]] = [attributes[1:]]
                # print(attributes)
        ''' for i in dict:
            print(i, dict[i]) '''
        final_dict = {}
        for i in dict:
            temp = dict[i]
            valid, invalid = 0, 0
            c1, c2 = 0, 0
            for per in temp:
                valid += round(float(per[0])*float(per[1]), 4)
                c1 += int(per[0])
                if per[2] != 'none':
                    invalid = round(float(per[2])*float(per[3]), 4)
                    c2 += int(per[2])
            final_dict[i] = [c1, round(valid/c1, 4)]
            if c2 != 0:
                final_dict[i].extend([c2, round(invalid/c2, 4)])

        final_list = sorted(dict, key=lambda x: [final_dict[x][1]])

        final = open(finalFilename, "w")
        for i in final_list:
            string = i+"\t"+"\t".join([str(i) for i in final_dict[i]])
            #print('s', string)
            final.write(string+'\n')

        return final_list


#sol = Aggregate_best()
#final_list = sol.find_average_dict("outputs/6_letter/output_100_", l_100)
# print(*final_list)