from random import randrange cislo = randrange(0, 3) # číslo je 0, 1, nebo 2 if cislo == 0: print('Kolečko') elif cislo == 1: print('Čtvereček') else: # tady musí být číslo 2 print('Trojúhelníček')