rectangular number pattern in python