Master Python: Use For And If In One Line!