Python:取近似值四舍五入

def ab():
    a_str = str(input())
    xiaoshu = a_str.split('.')[1]
    fir_xiao = list(xiaoshu)[0]
    if int(fir_xiao) <5:
        return int(a_str.split('.')[0])
    else:
        return int(a_str.split('.')[0])+1
print(ab())