有没有更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

return m and m.group(1)

这是一种很符合Python风格的做法。

如果m是None（或者其他被认为是“假”的东西），那么就返回m；但如果m是“真”的话，就返回m.group(1)。

Answer 2

另一种选择是使用：

return m.group(1) if m else m

这个方法很明确，你不需要费脑筋去理解它 :)

Answer 3

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