-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathclassifyByAppId.py
More file actions
74 lines (64 loc) · 1.71 KB
/
classifyByAppId.py
File metadata and controls
74 lines (64 loc) · 1.71 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
#!/bash/bin/python
import os
import csv
def readRawData():
res = []
for line in open("appid.txt"):
a = line.split("\n")
res.append(a[0])
return res
def readOneCVS(cvsFile):
res = []
#print cvsFile
i = 0
for line in open(cvsFile):
i = i + 1
if(i <= 1):#skip line 0 & line 1
continue
line = line.strip('\n')
a = line.split(",")
#print a
res.append(a)
return res
def oneDayData(d, qqiData):
c1 = 0
c2 = 0
for a in qqiData:
c1 += int(a[1])
c2 += int(a[2])
return (d, c1, c2)
def classifyALL():
QQiAppID = readRawData()
qqiData = []
qqData = []
qqiResult = []
qqResult = []
for filename in os.listdir('raw'):
print filename
pname = filename.split('.')
print pname
dataList = readOneCVS('raw/' + filename)
#print dataList
qqiData = [i for i in dataList if i[0] in QQiAppID]
qqData = [i for i in dataList if i[0] not in QQiAppID]
#print qqiData
#print qqData
#print "c1 " + str(c1)
#print "c2 " + str(c2)
qqiResult.append(oneDayData(pname[0],qqiData))
qqResult.append(oneDayData(pname[0], qqData))
#print result
writeResult('qqi.csv', qqiResult)
#print qqData
writeResult('qq.csv', qqResult)
def writeResult(result_csv, dataList):
if not os.path.exists('result'):
os.mkdir('result')
filePath = 'result/' + result_csv
writer = csv.writer(file(filePath, 'wb'))
for e in dataList:
writer.writerow(e)
#print filename
classifyALL()
#a = readRawData()
#print a