[Tutor] Tweaking list comprehensions

[Kent Johnson]

Kent Johnson wrote:
> Liam Clarke wrote:
> 
>> Hello,
>> I am fine tuning list comprehensions (at least my understandng
>> thereof), and I'm not near a Python interpreter at the moment, so I
>> was wondering if someone could tell me if I did OK -
>>
>> def approachA(files):
>>
>>     isHTML = []
>>     for filename in files:
>>         if filename.endswith('.htm') or filename.endswith('.html'):
>>             isHTML.append(filename)
>>     return isHTML
>>
>> def approachB(files):
>>
>>     isHTML = [filename if filename.endswith('.htm') or\
>>            filename.endswith(.html') for filename in files]     return 
>> isHTML
> 
> 
> Looks good to me.

Oops. Tony is right.
isHTML = [filename for filename in files if filename.endswith('.htm') or\
    filename.endswith('.html')]


Kent