Advertisement
Guest User

Untitled

a guest
Apr 23rd, 2017
58
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.00 KB | None | 0 0
  1. #load files line by line into memory and use recursive list comprehension. written in 30 min.
  2.  
  3. pwnall@pwnall-desktop:~$ cat fbtest2.py
  4. #!/usr/bin/env python
  5.  
  6. def open_csv(fd):
  7. data=[]
  8. with open(fd, 'r') as f:
  9. for line in f:
  10. row=line.split('\n')
  11. s=[s.lstrip() for r in row for s in r.split(',') if len(r) >2 and not s.startswith('NAME')]
  12. data.append(s)
  13. return data
  14.  
  15.  
  16. def speed(stride,leg):
  17. g=9.8
  18. speed=((( stride / leg) -1) * (leg*g)**.5)
  19. return speed
  20.  
  21. def compare(data1, data2):
  22. dinos=[]
  23. for s in data1:
  24. sname=s[0]
  25. leg=s[1]
  26. for d in data2:
  27. dname=d[0]
  28. stride=d[1]
  29. if sname==dname:
  30. dino_speed=speed(float(stride), float(leg))
  31. dinos.append((sname, dino_speed))
  32. return dinos
  33.  
  34. data1=open_csv('data1.csv')
  35. data2=open_csv('data2.csv')
  36.  
  37. dinos_sorted=sorted(compare(data1, data2),key=lambda x:x[1], reverse=True)
  38.  
  39. for dino in dinos_sorted:
  40. print dino[0]
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement