人到中年有點(diǎn)甜
2021-09-25 13:23:41
創(chuàng)建一個(gè) 4 x 4 矩陣并在父對(duì)角線(xiàn)上方添加值 4,5,6。我無(wú)法理解如何實(shí)現(xiàn)此輸出。嘗試對(duì)角線(xiàn)法但失敗了。我想要如下所示的輸出。 [[0 4 0 0] [0 0 5 0] [0 0 0 6] [0 0 0 0]]
2 回答

侃侃無(wú)極
TA貢獻(xiàn)2051條經(jīng)驗(yàn) 獲得超10個(gè)贊
In [120]: np.diag([4,5,6],1)
Out[120]:
array([[0, 4, 0, 0],
[0, 0, 5, 0],
[0, 0, 0, 6],
[0, 0, 0, 0]])

米琪卡哇伊
TA貢獻(xiàn)1998條經(jīng)驗(yàn) 獲得超6個(gè)贊
我不知道這是否有效,但我發(fā)現(xiàn)了一種不同的、有點(diǎn)復(fù)雜的方法來(lái)實(shí)現(xiàn)相同的目標(biāo),只是想與你們分享。(我是 Python 初學(xué)者):
mat1 = np.zeros((4,4))
list1 = [4,5,6]
for i in range(mat1.shape[1]):
for j in range(mat1.shape[0]):
if i<len(list1) and i==j:
mat1[i][j+1] = list1[i]
print(mat1)
添加回答
舉報(bào)
0/150
提交
取消