numpy.eye(N,M=None, k=0, dtype=<type 'float'>)

關注第一個第三個引數就行了

第一個引數：輸出方陣（行數=列數）的規模，即行數或列數

第三個引數：預設情況下輸出的是對角線全“1”，其餘全“0”的方陣，如果k為正整數，則在右上方第k條對角線全“1”其餘全“0”，k為負整數則在左下方第k條對角線全“1”其餘全“0”。

>>> np.eye(2, dtype=int)
array([[1, 0],
       [0, 1]])
>>> np.eye(3, k=1)
array([[ 0.,  1.,  0.],
       [ 0.,  0.,  1.],
       [ 0.,  0.,  0.]])
 

官方手冊：http://docs.scipy.org/doc/numpy/reference/generated/numpy.eye.html