codewars[7]-python Friend or Foe?

list中保留四字母的，而後return。

解

def friend(x):
i = len(x)
ii = []
a = 0
while a < i:
if len(x[a]) == 4:
ii.append(x[a])
a += 1
return ii

 

別人的超pythonic的題解。。

def friend(x):
return [f for f in x if len(f) == 4]