Numpyメモ(1):単位行列が欲しい

そんなときはidentity()関数を使えばよい。

>>>from numpy import *
>>>identity(2)
>>>array([[1.,0.],
         [0.,1.]])

同様のものにzeros()やones()がある。