Como calcular a média aritmética, ponderada e harmônica em Python - Escreva um programa Python que receba as 3 notas - Lista de Exercícios Resolvidos de Python

Primeira nota: 6.7
Segunda nota: 5.4
Terceira nota: 9.5
Informe uma letra [S, P, H]: P
A média do aluno é: 6.87

----------------------------------------------------------------------
Se precisar de ajuda com o código abaixo, pode me chamar
no WhatsApp +55 (62) 98553-6711 (Osmar)
----------------------------------------------------------------------

# função principal do programa
def main():
  # vamos pedir para o usuário informar as notas
  n1 = float(input("Primeira nota: "))
  n2 = float(input("Segunda nota: "))
  n3 = float(input("Terceira nota: "))
    
  # agora vamos pedir uma letra
  letra = input("Informe uma letra [S, P, H]: ")[0]
    
  # é media aritmética?
  if (letra == 'S') or (letra == 's'):
    soma = n1 + n2 + n3
    media = soma / 3.0
    # mostra a média
    print("A média do aluno é: {0}".format(media))
  # é media ponderada?
  elif (letra == 'P') or (letra == 'p'):
    media = (n1 * (50.0 / 100.0)) + (n2 * (30.0 / 100.0)) + \
      (n3 * (20.0 / 100.0))
    # mostra a média
    print("A média do aluno é: {0}".format(media))
  # é media harmônica?
  elif (letra == 'H') or (letra == 'h'):
    soma = (1.0 / n1) + (1.0 / n2) + (1.0 / n3)
    media = 3 / soma
    # mostra a média
    print("A média do aluno é: {0}".format(media))
  else:
    print("Letra inválida.")
  
if __name__== "__main__":
  main()