对角线为矩阵阶的倍数的恒等矩阵

Question

如何编写python程序以获取对角线为矩阵顺序倍数的单位矩阵。可以选择接受用户输入

Answer 1

import numpy as np

...
n = input("Enter size:")
matrix = n*np.identity(n)
...

Answer 2

这可以通过多种方式实现。

使用嵌套循环

mat_ord = int(input('Matrix Order: '))

mat = []

for i in range(mat_ord):
    row = []
    for j in range(mat_ord):
        row.append(1) if i == j else row.append(0)
        
    mat.append(row)
    
print(*mat, sep = '\n')

使用列表理解

mat_ord = int(input('Matrix Order: '))

mat = [
    [
        1 if i == j else 0 for i in range(mat_ord)
    ]
    for j in range(mat_ord)
]

print(*mat, sep = '\n')

我把它留给你去找谁。关于倍数，由于尚不清楚您希望倍数是多少，因此只需将1替换为x * mat_ord，其中x是您希望倍数的倍数是。