Generate Array In Python Interview