Python コードの簡略化

Question

pythonのコードの簡略化の方法で分からないところがあります。

words = ['cat','dog','bird','rabbit']

def test(words):
    for word in words:
        if len(word) == 4:
            return word

#####仕様
長さが４のword（ここではbird）をreturnする関数です。
上記の関数を簡略化することはできるでしょうか？
できればワンライナーで書きたいのですができますか？

Answer 1

質問でのコードのように

• 最初に見つかった要素だけを返す
• 見つからない場合はNoneを返す

のであれば、next() を使うのがよいと思います。

words = ['cat','dog','bird','rabbit']
ret = next((d for d in words if len(d)==4),None)
print(ret)

Answer 2

これが一番シンプルだと思います。

words = ['cat','dog','bird','rabbit']
ret = ([d for d in words if len(d)==4] or [None])[0]
print(ret)

Answer 3

p.py

words = ['cat', 'dog', 'bird', 'rabbit']
w4 = (list(filter(lambda w: len(w) == 4, words)) + [None])[0]
print(repr(w4))


words = ['cat', 'dog', 'rabbit']
w4 = (list(filter(lambda w: len(w) == 4, words)) + [None])[0]
print(repr(w4))

実行結果

$ python3 p.py
'bird'
None

Answer 4

test=lambda words : [word for word in words if len(word)==4][0]