Python Single Line If Assignment