1. 程式人生 > 程式設計 >python實現迴旋矩陣方式(旋轉矩陣)

python實現迴旋矩陣方式(旋轉矩陣)

我們知道Python中是沒有陣列 這種資料結構的,所以要想實現迴旋矩陣,需要先匯入一個numpy包, 它是一個由多維陣列物件和用於處理陣列的例程集合組成的python擴充程式庫,可以用來做陣列算數和邏輯運算

思路:

首先建立一個n*m全為零的矩陣,然後按照旋轉規律依次替換裡面的值

最外層的迴圈代表替換一圈資料,這一圈資料在陣列中的形狀是一個矩形,我們分析可以知道,最後一輪迴圈有兩種情況:

1、替換一圈(矩形)資料

2、替換一個數據

如果是第一種情況,我們讓迴圈正常執行就可以了,但是如果是第二種,就需要加一個if語句進行判斷。因為最後一輪大迴圈只需要替換一個數據,那麼我們只需要讓它走完第一個小迴圈最後一個元素必然被替換,所以只需要在第一個小迴圈後判斷一下替換的最後一個元素是否等於理論上最後一個元素的值即可

import numpy as np
def rotary_matrix(n,m):
 array=np.zeros((n,m))
 #起始點
 x=y=0
 ret=array[x][y]=1
 #設定一個變數i用來在迴圈中判斷是否需要旋轉矩陣
 i = 0
 while ret < n*m:
  #從上邊一行開始由左到右新增元素
  while y<m-1-i:
   y+=1
   ret+=1
   array[x][y]=ret
   
  #若ret等於最後一個元素的值,則表示所有元素新增完畢,這時候可以退出迴圈
  if ret == n * m:
   break
   
  # 從右邊一列開始由上到下新增元素
  while x<n-1-i:
   x+=1
   ret+=1
   array[x][y] = ret
  # 從下邊一行開始由右到左新增元素
  while y>0+i:
   y-=1
   ret+=1
   array[x][y]=ret
  # 從左邊一列開始由下到上新增元素
  while x>0+1+i:
   x-=1
   ret+=1
   array[x][y] = ret
   
  i+=1
 print(array)

以上這篇python實現迴旋矩陣方式(旋轉矩陣)就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。