默认功能参数基于先前的参数

Question

>>> def foo( a, b=a+1 ):
...     return b
...
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
NameError: name 'a' is not defined

我知道的解决方法是：

>>> def foo( a, b=None ):
...     if b == None:
...         b = a + 1
...     return b
... 
>>> foo(5)
6

我的用例：我需要b是a形状的单位矩阵。可以更优雅地做到这一点，还是“正确的方法”？