Python: Remove division decimal

Question

I have made a program that divides numbers and then returns the number, But the thing is that when it returns the number it has a decimal like this:

2.0
         


        

Answer 1

def division(a, b):
    return a / b if a % b else a // b