numpy.matlib.eye()
— another function to do matrix operations in numpy. Returns a matrix with diagonal ones and zeros elsewhere.
Syntax: numpy.matlib.eye (n, M = None, k = 0, dtype = ’ float ’, order =’ C ’)
Parameters:
n: [int] Number of rows in the output matrix.
M: [int, optional] Number of columns in the output matrix, defaults is n.
k: [int, optional] Index of the diagonal. 0 refers to the main diagonal, a positive value refers to an upper diagonal, and a negative value to a lower diagonal. Default is 0.
dtype: [optional] Desired output data-type .
order: Whether to store multi-dimensional data in row-major (C-style) or column-major (Fortran-style) order in memory.Return: A nx M matrix where all elements are equal to zero, except for the k-th diagonal, whose values are equal to one.
Code # 1:
|
Output:
Output matrix: [[1.0.0 .] [0. 1. 0.] [0. 0. 1.]]
Code # 2:
|
Output:
Output matrix: [[0 1 0 0 0] [0 0 1 0 0] [0 0 0 1 0] [0 0 0 0 1]]