diff --git a/src/codewars/kata_who_the_most_money.py b/src/codewars/kata_who_the_most_money.py new file mode 100644 index 0000000..a521421 --- /dev/null +++ b/src/codewars/kata_who_the_most_money.py @@ -0,0 +1,15 @@ +class Student: + def __init__(self, name, fives, tens, twenties): + self.name = name + self.fives = fives + self.tens = tens + self.twenties = twenties + + +def most_money(students): + student_names = [x.name for x in students] + student_money = [x.fives * 5 + x.tens * 10 + x.twenties * 20 for x in students] + + if len(set(student_money)) == 1 and len(student_money) > 1: + return "all" + return max(list(zip(student_names, student_money)), key=lambda x: x[1])[0]