python - 更pythonic的方式来写这个？

Question

我这里有这段代码:

import re
def get_attr(str, attr):
    m = re.search(attr + r'=(\w+)', str)
    return None if not m else m.group(1)

str = 'type=greeting hello=world'

print get_attr(str, 'type')   # greeting    
print get_attr(str, 'hello')  # world
print get_attr(str, 'attr')   # None

哪个有效，但我不是特别喜欢这一行:

return None if not m else m.group(1)

在我看来，如果我们可以使用三元运算符，这看起来会更清晰:

return (m ? m.group(1) : None)

但这当然不存在。你有什么建议？

Answer 1

Python 有一个三元运算符。你正在使用它。它只是采用 X if Y else Z 形式。

也就是说，我倾向于把这些东西写出来。如果您牺牲了清晰度，那么将东西放在一条线上并不是很好。

def get_attr(str, attr):
    m = re.search(attr + r'=(\w+)', str)
    if m:
        return m.group(1)

    return None