python - 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]