Guest User

Untitled

a guest
Oct 17th, 2018
95
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.12 KB | None | 0 0
  1. import sys,codecs
  2. '''
  3. format emiter receiver contacts
  4.  
  5. example: 5767599 242682935 4
  6.  
  7. '''
  8. links={}
  9. i=0
  10. f=file("/dev/stdin")
  11.  
  12.  
  13. for line in f:
  14. i+=1
  15. #print line
  16. line=line.split(";");
  17. emiter=int(line[0])
  18. reciever=int(line[1])
  19. contacts=int(line[2])
  20. l=sorted([emiter,reciever])
  21. #rel_key='-'.join(map(str,l))
  22. rel_key=(l[0],l[1])
  23. di=0 if caller<callee else 1
  24. if not rel_key in links:
  25. links[rel_key]=[0,0]
  26. links[rel_key][di]+=contacts
  27. if i%1000000==0:
  28. print "read %i million contacts"%(i/1000000)
  29. print len(links),"relationships in dict"
  30.  
  31. f.close()
  32. f=codecs.open("network-mutual-graph","w")
  33. f2=codecs.open("network-directed-graph","w")
  34.  
  35. for link,contacs in links.iteritems():
  36. link=link.split('-')
  37. if contacts[0]==0:
  38. f2.write("%s %s %i \n"%(link[1],link[0],contacts[1]))
  39. elif contacts[1]==0:
  40. f2.write("%s %s %i \n"%(link[0],link[1],contacts[0]))
  41. else:
  42. f.write("%s %s %i %i \n"%(link[0],link[1],contacts[0],contacts[1]))
Add Comment
Please, Sign In to add comment