python for in if one line