我想根據另一個矩陣原始大小 ROW x COLS 創建一個矩陣 ROW x COLS。
ROWS = len(mat)
COLS = len(mat[0])
res = [[0 for i in range(ROWS)] for i in range(COLS)]
上面的代碼不適用于以下邊緣情況:
mat = [[3],[4],[5],[5],[3]]
我想要的輸出是:
[[0],[0],[0],[0],[0]]
但是,我得到的是:
[[0,0,0,0,0]]
如何調整我的代碼以適應任何情況?(我不想使用 numpy 或任何其他庫)
uj5u.com熱心網友回復:
像這樣交換 ROWS 和 COLS:
res = [[0 for i in range(COLS)] for i in range(ROWS)]
uj5u.com熱心網友回復:
嘗試:
>>> [[0 for i in range(len(mat[0]))] for j in range(len(mat))]
[[0], [0], [0], [0], [0]]
最好使用不同的回圈變數而不是i兩者:
轉載請註明出處,本文鏈接:https://www.uj5u.com/qianduan/466047.html
上一篇:按多個分隔的數字對字串進行排序
