-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathchess_categorize.py
More file actions
112 lines (82 loc) · 2.47 KB
/
chess_categorize.py
File metadata and controls
112 lines (82 loc) · 2.47 KB
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
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
import sys
filename = sys.argv[1]
base_sets = []
set_names = []
sets = [[] for i in range(129)]
name_base = 's'
col_states = ['a', 'b', 'c', 'd', 'e', 'f', 'g', 'h']
row_states = ['1', '2', '3', '4', '5', '6', '7', '8']
target_states = ['zero' ,'one', 'two', 'three', 'four', 'five', 'six', 'seven', 'eight', 'nine', 'ten', 'eleven', 'twelve', 'thirteen', 'fourteen', 'fifteen', 'sixteen']
for i in range(7):
base_sets.append(name_base+str(i))
for i in range(6):
s = base_sets[i]
if(i%2 == 0):
for item in col_states:
set_names.append(s+item)
else:
for item in row_states:
set_names.append(s+item)
for item in target_states:
s = base_sets[6]
set_names.append(s+item)
#print('set_names:', set_names)
for i in range(len(set_names)):
print('i: ', i , ':', set_names[i])
exit(1)
reader = open(filename, 'r')
raw = reader.readlines()
data = []
for item in raw:
data.append(item.strip("\n").split(","))
print(len(data))
transpose = []
for s in range(7):
col = []
for i in range(len(data)):
col.append(data[i][s])
transpose.append(col)
#print(set_names)
for column in range(7):
for state in range(8):
for i in range(len(data)):
#print('i:', i, 'state:', transpose[column][i])
if(column%2 == 0):
if(transpose[column][i] == col_states[state]):
sets[(column*3)+state].append(i)
else:
if(transpose[column][i] == col_states[state]):
sets[(column*3)+state].append(i)
specials = [6]
for special in specials:
for state in range(17):
for i in range(len(data)):
if(transpose[special][i] == target_states[state]):
sets[(special*8)+state].append(i)
#print(sets[127])
write_name = filename + "_settled.txt"
writer = open(write_name, 'w+')
#writer. write(str(len(data)))
#writer.write("\t")
writer.write("65")
writer.write("\n")
for i in range(65):
writer.write(str(len(sets[i])))
for val in range(len(sets[i])):
writer.write("\t")
writer.write(str(sets[i][val]))
writer.write("\n")
'''
for i in range(129):
print('Length of set', i , ':', len(sets[i]))
for i in range(len(data)):
for j in range(128):
try:
writer.write(str(sets[j][i]))
except:
print('j:', j, 'i:', i)
exit(1)
writer.write("\t")
writer.write(str(sets[128][i]))
writer.write("\n")
'''