Opening_Weekend/genres.py at master · SwammyD/Opening_Weekend · 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
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
import csv

#Action,Adventure,Animation,Biography,Comedy,Crime,Documentary,Drama,Family,Fantasy,History,Horror,Music,Musical,Mystery,Romance,Sci-Fi,Sport,Thriller,War,Western

with open('new_out_with_finance.csv','r') as tsvin, open('fixed.csv', 'w') as csvout:
	tsvin = csv.reader(tsvin, delimiter=',')
	csvout = csv.writer(csvout)
	for row in tsvin:
		if "Action" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Adventure" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Animation" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Biography" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Comedy" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Crime" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Documentary" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Drama" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Family" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Fantasy" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "History" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Horror" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Music" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Musical" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Mystery" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Romance" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Sci-Fi" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Sport" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Thriller" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "War" in row[8]:
			row.append(1)
		else:
			row.append(0)
		if "Western" in row[8]:
			row.append(1)
		else:
			row.append(0)

		row.pop(8)

		csvout.writerows([row])