How to create identity matrix with numpy

前端 未结 3 2544
走了就别回头了
走了就别回头了 2021-02-20 08:39

How do I create an identity matrix with numpy? Is there a simpler syntax than

numpy.matrix(numpy.identity(n))
3条回答
  •  情话喂你
    2021-02-20 09:15

    Also np.eye can be used to create an identity array (In).

    For example,

    >>> 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.]])
    

提交回复
热议问题