我必须计算一系列数据中出现的节点对的次数和百分比我能很好地计算这个,但是,当我还包括第三个参数时,问题就来了,即以秒为单位的时间。这些节点对彼此接触并花费一些时间连接,时间以秒为单位还应计算到相应节点对的总时间(秒)。
例:输入
Node Node Time
A B 455
A B 456
A B 463
A C 4
A C 675
C B 64
C B 78
C B 579
到目前为止,我得到的输出是正确的。
Node Node paircount pairpercentage
A B 3 37.5
A C 2 25
C B 3 37.5
我应该得到的结果是
Node Node Paircount pairpercentage Time
A B 3 37.5 1374
A C 2 25 679
C B 3 37.5 721
代码:
from collections import defaultdict
d = defaultdict(int)
# get number of occurences for the first two columns
with open('Inputfile.txt', 'r') as f:
#f.readline() # discard the header line
for numlines, line in enumerate(f,1):
line = line.strip().split()
c = line[0], line[1]
d[c] += 1
# compute 100*(occurences/numlines) for each key in d
d = {k:(v, 100*float(v)/numlines) for k,v in d.iteritems()}
with open('outputfile.txt', 'w') as outfile:
for k in d:
#print k, d[k]
outfile.write("%s %s\n" % (k, d[k]))
注意:上面的代码非常适合我前面提到的一半,我需要一些帮助来帮助剩下的一半处理关于节点对的时间加法。
输入文件
5454 5070 2755.0
5070 4391 2935.0
1158 305 1.0
5045 3140 48767.0
4921 3140 58405.0
5372 2684 460.0
1885 1158 351.0
1349 1174 6375.0
1980 1174 650.0
1980 1349 650.0
4821 2684 469.0
4821 937 459.0
2684 937 318.0
1980 606 390.0
1349 606 750.0
1174 606 750.0
5045 3545 8133.0
4921 3545 8133.0
3545 3140 8133.0
5045 4243 14863.0
4921 4243 14863.0
4243 3545 8013.0
4243 3140 14863.0
4821 4376 5471.0
4376 937 136.0
2613 968 435.0
5372 937 83.0
代码2:在下面的答案的帮助下,我可以计算paircount和时间,但我现在无法得到百分比我也在寻找一些输出清洁。
from collections import defaultdict
paircount = defaultdict(int)
pairtime = defaultdict(float)
pairper = defaultdict(float)
#get number of pair occurrences and total time
with open('USC_Test.txt', 'r') as f:
with open('pair.txt', 'w') as o:
numline = 0
for line in f:
numline += 1
line = line.split()
pair = line[0], line[1]
paircount[pair] += 1
pairtime[pair] += float(line[2])
pairper = float(paircount/line)*100
print "%s\n" % paircount
print "%s\n" % pairtime
print "%s\n" % pairper
输出:仅当
pairper = float(paircount/line)*100
被注释时,否则会出现如下所示的错误。defaultdict(<type 'int'>, {('1349', '606'): 1, ('2684', '937'): 1,
defaultdict(<type 'float'>, {('1349', '606'): 750.0, ('2684', '937'): 318.0,
但当
pairper = float(paircount/line)*100
不在评论中时错误:
TypeError: unsupported operand type(s) for /: 'collections.defaultdict' and 'list'
预期输出没有文本
defaultdict(<type 'int'>, or deafultdict(<type 'float'>
,但只有node node paircount pairper pairtime
1349 606 1 somevalue 750.0
2684 937 1 somevalue 318.0
任何建议谢谢。
最佳答案
您可以使用另一个defaultdict
来将时间相加。(编辑:现在进行其他更改。)
from collections import defaultdict
paircount = defaultdict(int)
pairtime = defaultdict(float)
with open('Inputfile.txt') as f:
numlines = 0
for line in f:
numlines += 1
line = line.split()
pair = line[0], line[1]
paircount[pair] += 1
pairtime[pair] += float(line[2])
pairper = dict((pair, c * 100.0 / numlines) for (pair, c) in paircount.iteritems())
for pair, c in paircount.iteritems():
print pair[0], pair[1], c, pairper[pair], pairtime[pair]
我还修复了一个涉及
numlines
的off错误,删除了多余的.strip()
,并为清晰起见重命名了一些变量。