Bryting-NN-2

import numpy as np

def Uvi(sinv,n1,n2):
    return(np.arcsin(n1*sinv/n2)*180/np.pi)

Ivi = float(input("Skriv inn innfallsvinkel i grader: "))
sinv = np.sin(Ivi*np.pi/180)

Mvalg = float(input("Kva medium går lyset ut frå (skriv inn talet)?\n 1-Vatn\n 2-Kronglas\n \
3-Flintglas\n 4-Diamant\n 5-Luft\n Ditt val: "))
M2valg = float(input("Kva medium går lyset inn i (skriv inn talet)?\n 1-Vatn\n 2-Kronglas\n \
3-Flintglas\n 4-Diamant\n 5-Luft\n Ditt val: "))

m1 = 1.333
m2 = 1.55
m3 = 1.7
m4 = 2.287
m5 = 1

if(Mvalg==1):
    n1 = m1
elif(Mvalg==2):
    n1 = m2
elif(Mvalg==3):
    n1 = m3
elif(Mvalg==4):
    n1 = m4
else:
    print("Du skreiv inn noko ugyldig.")

if(M2valg==1):
    n2 = m1
elif(M2valg==2):
    n2 = m2
elif(M2valg==3):
    n2 = m3
elif(M2valg==4):
    n2 = m4
else:
    print("Du skreiv inn noko ugyldig.")

print("Brytingsvinkelen er ",round(Uvi(sinv,n1,n2),2),"grader.")