Brytning-BM-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("Hvilket medium går lyset ut fra (skriv inn tallet)?\n 1-Vann\n 2-Kronglass\n \
3-Flintglass\n 4-Diamant\n 5-Luft\n Ditt valg: "))
M2valg = float(input("Hvilket medium går lyset inn i (skriv inn tallet)?\n 1-Vann\n 2-Kronglass\n \
3-Flintglass\n 4-Diamant\n 5-Luft\n Ditt valg: "))

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 skrev inn noe ugyldig.")

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

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