Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- #load files line by line into memory and use recursive list comprehension. written in 30 min.
- pwnall@pwnall-desktop:~$ cat fbtest2.py
- #!/usr/bin/env python
- def open_csv(fd):
- data=[]
- with open(fd, 'r') as f:
- for line in f:
- row=line.split('\n')
- s=[s.lstrip() for r in row for s in r.split(',') if len(r) >2 and not s.startswith('NAME')]
- data.append(s)
- return data
- def speed(stride,leg):
- g=9.8
- speed=((( stride / leg) -1) * (leg*g)**.5)
- return speed
- def compare(data1, data2):
- dinos=[]
- for s in data1:
- sname=s[0]
- leg=s[1]
- for d in data2:
- dname=d[0]
- stride=d[1]
- if sname==dname:
- dino_speed=speed(float(stride), float(leg))
- dinos.append((sname, dino_speed))
- return dinos
- data1=open_csv('data1.csv')
- data2=open_csv('data2.csv')
- dinos_sorted=sorted(compare(data1, data2),key=lambda x:x[1], reverse=True)
- for dino in dinos_sorted:
- print dino[0]
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement