How to return all list elements of a given length?

Question

I'm trying to return words that have a specific length.

This is my code so far. words is a list and size is a positive integer.

def by_size(words, size)
    for word in words:
        if len(word) == size:

I'm not sure how to continue. by_size(['a', 'bb', 'ccc', 'dd'], 2) should return ['bb', 'dd']. Any suggestions would be of great help.

Answer 1

I would use a list comprehension:

def by_size(words, size):
    return [word for word in words if len(word) == size]

Answer 2

def by_size(words,size):
    result = []
    for word in words:
        if len(word)==size:
            result.append(word)
    return result

Now call the function like below

desired_result = by_size(['a','bb','ccc','dd'],2)

where desired_result will be ['bb', 'dd']