로또 당첨 프로그램

자바에서 만들었던 로또 당첨 프로그램을 똑같이 파이썬에서 구현해보았다.

import random
lottoNum = [0,0,0,0,0,0]

    

for i in range(6):
    count=0
    randomNum=(random.randint(1,45))
    for j in range(6):
        if lottoNum[j] == randomNum:
            count+=1
    if count == 0:
        lottoNum[i] = randomNum
    else:
        i-=1
        
choice = int(input('출력을 원하는 등수:'))

if choice == 1:
    print('------1등------')
    for i in range(6):
        print(lottoNum[i], end='\t')
    print()
elif choice ==2:
    print('------2등------')
    for i in range(6):
        for j in range(6):
            if j==i:
                print('X',end='\t')
            else:
                print(lottoNum[j],end='\t')
        print()
elif choice ==3:
    print('------3등------')
    for i in range(6):
        for j in range(i+1,6):
            for h in range(6):
                if i==h:
                    print('X',end='\t')
                elif j==h:
                    print('X',end='\t')
                else:
                    print(lottoNum[h],end='\t')
            print()
elif choice ==4:
    print('------4등------')
    for i in range(6):
        for j in range(i+1,6):
            for k in range(j+1,6):
                for h in range(6):
                    if j==h:
                        print('X',end='\t')
                    elif i==h:
                        print('X',end='\t')
                    elif k==h:
                        print('X',end='\t')
                    else:
                        print(lottoNum[h],end='\t')
                print()
                    
    
elif choice ==5:
    print('------5등------')
    for i in range(6):
        for j in range(i+1,6):
            for k in range(j+1,6):
                for q in range(k+1,6):
                    for h in range(6):
                        if j==h:
                            print('X',end='\t')
                        elif i==h:
                            print('X',end='\t')
                        elif k==h:
                            print('X',end='\t')
                        elif q==h:
                            print('X',end='\t')
                        else:
                            print(lottoNum[h],end='\t')
                    print()