numpy里生成单位矩阵？

  数学 线性代数 Python    浏览次数：3175        分享
二维码
手机扫描二维码
  机器学习面试也能刷题？
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里矩阵乘法matmul，@和dot的区别？   1回答

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

怎么对numpy array转置？   2回答

numpy怎么进行矩阵乘法   1回答

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

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

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

能直接把csv格式的数据文件读到numpy array里吗？   1回答

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

怎么用numpy生成一个三维的tensor？   1回答

numpy.array从行向量转为列向量   3回答

求一个nxn的numpy array的对角线的和？   1回答

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

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

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

我们谢绝答非所问。

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

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