-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathgetTopic.py
117 lines (93 loc) · 2.79 KB
/
getTopic.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
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
113
114
115
116
117
from collections import defaultdict
import os
import pandas as pd
cwd = os.getcwd()
path = str(cwd) + "\\files\\RepliesWithTopicCopy.csv"
df = pd.read_csv(path, header=None,low_memory=False)
tweetReplyDict = defaultdict(list)
count = 0
for index, row in df.iterrows():
if(index == 0):
continue
# print(row[0])
if(row[4] == "1"):
row_list = []
row_list.append(row[0])
row_list.append(row[1])
row_list.append(row[2])
row_list.append(row[3])
if (row[3][0] == '-'):
row_list.append("S" + row[3])
else:
row_list.append("S+" + row[3])
count += 1
row_list.append('T1')
tweetReplyDict[count] = row_list
if (row[5] == "1"):
row_list = []
row_list.append(row[0])
row_list.append(row[1])
row_list.append(row[2])
row_list.append(row[3])
if (row[3][0] == '-'):
row_list.append("S" + row[3])
else:
row_list.append("S+" + row[3])
count += 1
row_list.append('T2')
tweetReplyDict[count] = row_list
if (row[6] == "1"):
row_list = []
row_list.append(row[0])
row_list.append(row[1])
row_list.append(row[2])
row_list.append(row[3])
if (row[3][0] == '-'):
row_list.append("S" + row[3])
else:
row_list.append("S+" + row[3])
count += 1
row_list.append('T3')
tweetReplyDict[count] = row_list
if (row[7] == "1"):
row_list = []
row_list.append(row[0])
row_list.append(row[1])
row_list.append(row[2])
row_list.append(row[3])
if (row[3][0] == '-'):
row_list.append("S" + row[3])
else:
row_list.append("S+" + row[3])
count += 1
row_list.append('T4')
tweetReplyDict[count] = row_list
if (row[8] == "1"):
row_list = []
row_list.append(row[0])
row_list.append(row[1])
row_list.append(row[2])
row_list.append(row[3])
if (row[3][0] == '-'):
row_list.append("S" + row[3])
else:
row_list.append("S+" + row[3])
count += 1
row_list.append('T5')
tweetReplyDict[count] = row_list
if (row[9] == "1"):
row_list = []
row_list.append(row[0])
row_list.append(row[1])
row_list.append(row[2])
row_list.append(row[3])
if (row[3][0] == '-'):
row_list.append("S" + row[3])
else:
row_list.append("S+" + row[3])
count += 1
row_list.append('T6')
tweetReplyDict[count] = row_list
print(index)
df_clean = pd.DataFrame.from_dict(tweetReplyDict,orient='index')
df_clean.to_csv('files\\RepliesFormatted.csv',header=False)