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

# 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))