码迷,mamicode.com
首页 > 编程语言 > 详细

python练习--生成螺旋矩阵

时间:2020-01-19 18:55:01      阅读:81      评论:0      收藏:0      [点我收藏+]

标签:元素   ros   螺旋矩阵   往里面   思路   nump   练习   rect   下一步   

 螺旋矩阵,像下面这样,看了就能理解,不多解释了。

[[ 1.  2.  3.  4.  5.  6.]
 [20. 21. 22. 23. 24.  7.]
 [19. 32. 33. 34. 25.  8.]
 [18. 31. 36. 35. 26.  9.]
 [17. 30. 29. 28. 27. 10.]
 [16. 15. 14. 13. 12. 11.]]

说一下我的思路:
1.先生成一个零矩阵,再往里面填充数字;
2.每填充完一个,都判断下一步该往哪走,下一步有四种状态:横+、纵+、横-、纵-,依次循环。
3.每种状态间切换的条件:原状态的下一步的元素已被填充(!=0)或索引溢出。

下面是一种具体实现(上面只是整体思路,与实现并不完全对应):
 1 import numpy as np
 2 def r_matrix(n):
 3     mt = np.zeros(n**2).reshape(n,n)
 4     x, y = 0, 0
 5     num = 1
 6     direction = 1
 7     
 8     while mt[x][y] == 0:
 9         mt[x][y] = num
10         num += 1
11         if -1<y+direction<n  and mt[x][y+direction] == 0:
12             y += direction
13         elif x+direction<n and mt[x+direction][y] == 0:
14             x += direction
15         else:
16             direction = -direction
17             y += direction
18     return mt  

python练习--生成螺旋矩阵

标签:元素   ros   螺旋矩阵   往里面   思路   nump   练习   rect   下一步   

原文地址:https://www.cnblogs.com/pyonwu/p/12215086.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!