Нужно написать программу для вычисления сопротивления по закону ома

Python
z = str(input('\tЗакон Ома\nЩо шукаємо? I,R,U?: '))
if z == 'I':
u = int(input("U = "))
r = int(input("R = "))
print('I = ',u,'/',r,'= ',u/r,' A')
elif z == 'U':
i = int(input("I = "))
r = int(input("R = "))
print('I = ',i,'*',r,'= ',i*r,' B')
elif z == 'R':
u = int(input("U = "))
i = int(input("I = "))
print('I = ',u,'/',i,'= ',u/i,' Oм')