Advertisement
Guest User

Untitled

a guest
Aug 20th, 2019
72
0
Never
Not a member of Pastebin yet? Sign Up, it unlocks many cool features!
text 1.03 KB | None | 0 0
  1. import csv
  2. import math
  3.  
  4. g=9.8**2
  5.  
  6. class dinosaur:
  7. def __init__(self, name, leg_length, diet, stride_length, stance):
  8. self.name = name
  9. self.leg_length = leg_length
  10. self.diet = diet
  11. self.stride_length = stride_length
  12. self.stance = stance
  13. self.speed = self.getSpeed()
  14.  
  15. def getSpeed(self):
  16. return ((self.stride_length / self.leg_length) -1) * math.sqrt(self.leg_length * g)
  17.  
  18. def readCSV(file):
  19. reader = csv.DictReader(open(file))
  20. result = {}
  21. for row in reader:
  22. key = row.pop('NAME')
  23. result[key] = row
  24. return result
  25.  
  26. def getDeets(legdiet, stridestance):
  27. dinos = []
  28. ldd = readCSV(legdiet)
  29. ssd = readCSV(stridestance)
  30. for d in ldd:
  31. dinos.append(dinosaur(d, float(ldd[d]['LEG_LENGTH']), ldd[d]['DIET'], float(ssd[d]['STRIDE_LENGTH']), ssd[d]['STANCE']))
  32. return dinos
  33.  
  34. if __name__ == '__main__':
  35. dinos = sorted(getDeets("dataset1.csv", "dataset2.csv"), key=lambda x: x.speed, reverse=True)
  36. i = 0
  37. while i < len(dinos):
  38. if dinos[i].stance == "bipedal":
  39. print("%s" % dinos[i].name)
  40. i=i+1
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement