if condition in for loop in python