python – ¿Alguien sabe hacer el output como se pide?

Imaginen que tienen un ciclo en Python que prueba números según condiciones. Como input está “X” y “Y”. La cantidad de números que vas a ingresar al programa depende de “X”, y “Y” son los números ingresados, el programa devuelve uno o varios números (Z1,Z2,Z3, etc)según la condiciones que haya cumplido “Y”(Cada condición da un numero diferente). ¿Cómo le harían para que el programa imprima esos números hasta que hayas ingresado los “X” cantidad de números que se mencionaron y de la siguiente forma?
El output debe de quedar así(Por ejemplo)
Si X es igual a 3:

Z1 Z2

Z1 Z5

Z7 Z8 Z9

En cada línea están las condiciones que cumplieron Y1,Y2 y Y3 respectivamente, separados por espacios.
Les dejo el código con el que estoy intentando si sirve de algo, solo no sé como hacer que el output salga como se pide.

https://github.com/Sphoreas/CodeForces/blob/main/Kids%20at%20the%20pary.py