:param n: dimension for nxn matrix :type n: int :return: Identity matrix of shape [n, n] >>> identity(3) [[1, 0, 0], [0, 1, 0], [0, 0, 1]]
(n: int)
| 81 | |
| 82 | |
| 83 | def identity(n: int) -> list[list[int]]: |
| 84 | """ |
| 85 | :param n: dimension for nxn matrix |
| 86 | :type n: int |
| 87 | :return: Identity matrix of shape [n, n] |
| 88 | >>> identity(3) |
| 89 | [[1, 0, 0], [0, 1, 0], [0, 0, 1]] |
| 90 | """ |
| 91 | n = int(n) |
| 92 | return [[int(row == column) for column in range(n)] for row in range(n)] |
| 93 | |
| 94 | |
| 95 | def transpose( |