python for in if list