Calculadora de médias - CÓDIGO PYTHON

Faça um programa que calcula diferentes tipos de médias a partir de 3 notas de provas.
• ENTRADA:
1. Primeira nota
2. Segunda nota
3. Terceira nota
4. O tipo de média que será calculado. As opções são:
 Média aritmética
 Média ponderada*
 Média harmônica
*Considere os pesos 2, 3 e 4 para a média ponderada
• PROCESSAMENTO:
O programa calcula a média de acordo com a opção desejada
• SAÍDA:
1. Valor da média

###########################################

def mAritmetica(n1,n2,n3):
    media=(n1+n2+n3)/3
    return media

def mPonderada(n1,n2,n3):
    media=((2*n1)+(3*n2)+(4*n3))/9
    return media

def mHarmonica(n1,n2,n3):
    media=(3/((1/n1)+(1/n2)+(1/n3)))
    return media

n1=float(input("Entre com a primeira nota"))
n2=float(input("Entre com a segunda nota"))
n3=float(input("entre com a terceira nota"))

tipo=int(input("Digite \n 1 - para media aritmetica.\n 2 - para ponderada.\n 3 - media harmonica. \n :"))

if tipo!=1 and tipo!=2 and tipo!=3:
    print("Entrada invalida")
 
else:
    if tipo==1:
        m=mAritmetica(n1,n2,n3)
 
    elif tipo==2:
        m=mPonderada(n1,n2,n3)
 
    else:
        m=mHarmonica(n1,n2,n3)
     
         
    print("A media é {: .2f}".format(m))