Написать программу, которая бы по введенному номеру единицы измерения (1 — килограмм, 2 — миллиграмм, 3 — грамм, 4 — тонна, 5 — центнер) и массе М выдавала соответствующее значение массы в килограммах. Программу для питона

s = input().split()

if s[0] == '1':

   print(s[1])

elif s[0] == '2':

   print(float(s[1]) / 1000000)

elif s[0] == '3':

   print(float(s[1]) / 1000)

elif s[0] == '4':

   print(float(s[1]) * 1000)

else:

   print(float(s[1]) * 100)

#Вывод вещественных чисел, т.к килограммов может быть нецелое #число