python-selenium/DataProcessing.py at master · FunKeen/python-selenium · 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
importpandasaspd
importnumpyasnp

data=pd.read_csv('Knockout.csv')
pd.set_option('future.no_silent_downcasting', True)

data.replace('Play-off for third place', 0, inplace=True)
data.replace('Final', 1, inplace=True)
data.replace('Semi-final', 2, inplace=True)
data.replace('Quarter-final', 4, inplace=True)
data.replace('Round of 16', 8, inplace=True)

arr=np.array(data)
arr=sorted(arr, key=lambday: y[2])


defwinner(item):
re=item[4]
ifitem[8] <item[9] oritem[8] ==item[9] anditem[6][1] <item[7][1]:
re=item[5]
returnre


forxinarr:
x[-1] =winner(x)

pd.DataFrame(arr).to_csv('PreKnockout.csv', index=False)