numpy里生成单位矩阵？

  数学 线性代数 Python    浏览次数：23760        分享
二维码
手机扫描二维码
  机器学习面试也能刷题？
0

numpy里怎么生成一个n x n的单位矩阵？

 

一亩三分   2018-10-18 06:13


   2个回答 
6

np.identity(n)可以生成$n\times n$的单位矩阵

>>> np.identity(4)
array([[ 1.,  0.,  0.,  0.],
       [ 0.,  1.,  0.,  0.],
       [ 0.,  0.,  1.,  0.],
       [ 0.,  0.,  0.,  1.]])

或者也可以用np.diag生成对角阵

>>> np.diag([1] * 4)
array([[1, 0, 0, 0],
       [0, 1, 0, 0],
       [0, 0, 1, 0],
       [0, 0, 0, 1]])


SofaSofa数据科学社区DS面试题库 DS面经

wxw_pku   2018-10-18 08:50

3

最省事的是numpy.eye了，比numpy.identity还少打了几个字母，哈哈

np.eye(3)
array([[1., 0., 0.],
       [0., 1., 0.],
       [0., 0., 1.]])
SofaSofa数据科学社区DS面试题库 DS面经

何立诚   2019-03-25 05:55



  相关主题

怎么对numpy array转置？   2回答

numpy里矩阵乘法matmul，@和dot的区别？   1回答

怎么把numpy array转成scipy里的稀疏矩阵？   1回答

删除numpy array中指定的一列   1回答

如何在numpy array尾部增加一行   2回答

numpy中获得tensor的维数？   2回答

调用np.average()时报错TypeError: No loop matching   1回答

怎么把scipy.sparse.csr.csr_matrix转成numpy array   1回答

python numpy里array和matrix的区别？   1回答

怎么生成一个填满逻辑真（True）的numpy array？   1回答

怎么把numpy array中的inf换成0？   1回答

怎么获得numpy array每行最小值的index？   1回答

回答问题时需要注意什么?

我们谢绝在回答前讲“生动”的故事。

我们谢绝“这么简单，你自己想”、“书上有的，你认真看”这类的回答；如果你认为对方的提问方式或者内容不妥，你可以直接忽略该问题，不用进行任何作答，甚至可以对该问题投反对票。

我们谢绝答非所问。

我们谢绝自己不会、硬要回答。

我们感激每一个用户在编写答案时的努力与付出！