[英]Trying to rotate a matrix 90 degrees but fails to work
def rotate_ninety(matrix):
putin = []
temporary = []
row_size = len(matrix)
col_size = len(matrix[0])
ccount = col_size-1
putin_idx = 0
while ccount > -1:
for i in range(row_size):
temporary.insert(i, matrix[i][ccount])
putin.insert(putin_idx, temporary)
ccount = ccount -1
temporary.clear()
putin_idx += 1
print(putin)
return 0
所以,如果我有輸入
[1,2,3]
[4,5,6]
[7,8,9]
結果應該是:
[3,6,9]
[2,5,8]
[1,4,7]
但是,當我打印 putin 時,我得到 [[] [] []],一個全是空的數組,我不明白我的邏輯哪里出了問題。 我知道還有其他更有效的方法可以做到這一點,但我看不出我的代碼有什么問題
當您將list
分配(追加或插入)到變量(另一個列表)時,將分配或追加list object
。 因為在將temporary
列表添加到putin
列表然后清除temporary
列表之后,已添加的 object 也被刪除了。 您需要將temporary
列表的copy
添加到putin
,如下所示:
import copy
def rotate_ninety(matrix):
putin = []
temporary = []
row_size = len(matrix)
col_size = len(matrix[0])
ccount = col_size-1
putin_idx = 0
while ccount > -1:
for i in range(row_size):
temporary.append(matrix[i][ccount])
putin.append(copy.copy(temporary))
ccount = ccount -1
temporary.clear()
putin_idx += 1
print(putin)
return 0
matrix = [[1,2,3],[4,5,6],[7,8,9]]
rotate_ninety(matrix)
結果將是:
[[3, 6, 9], [2, 5, 8], [1, 4, 7]]
這是 numpy 的簡單方法:
matrix = np.arange(1,10).reshape((3,3))
rotated_90 = matrix.T[::-1]
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.