1142.PUM in python

This is the solution of this problem in python:

N=int(input())
j=1
for i in range(1,N+1):
    for k in range(j,j+3):
        print(k,end=" ")
    print("PUM")
    j=j+4