Hola Jose, te comparto otra solución, aunque no es la correcta para esta etapa en tu aprendizaje es SOLO para que la tengas en cuanta para el futuro y que sepas que en Python como en otros lenguajes a muchas formas de realizar una misma tarea. En este momento es mejor tener en cuenta la respuesta de xve, y aprendas los fundamentos de bucles, listas y de más.
Aquí la solución:
En el código de anterior hago uso de comprensión de listas y unpacking asterisk, que son caracteristicas muy potentes del lenguaje.
Como dije al principio, estos temas son un poco avanzados en Python por lo que no deberías tenerlos en cuenta, hasta no tener claro los fundamentos básicos del lenguaje.