python基础编程_28_开关问题,蛇形填数

import numpy as np

myinput=input().split()#n盏灯,k个人
myinput=list(map(int,myinput))
lights=np.zeros(myinput[0])
for i in range(1,myinput[1]+1):
    for j in range(1,myinput[0]+1):
        if j%i==0 and i==1:
            lights[j-1]=1
        elif j%i==0:
            lights[j-1]*=-1

    

for i in range(myinput[0]):
    if lights[i]==1:
        print(i+1,end=' ')#输出灯的编号
    

7 3
1 5 6 7 

import numpy as np
myinput=int(input())
n=myinput
myarray=np.zeros((n,n))
count=0
line=n
while(count<n*n):
    for i in range(n-line,line):
        count+=1
        myarray[i][line-1]=count
    for i in range(line-2,n-line-1,-1):
        count+=1
        myarray[line-1][i]=count    
    for i in range(line-2,n-line-1,-1):
        count+=1
        myarray[i][n-line]=count
    for i in range(n-line+1,line-1):
        count+=1
        myarray[n-line][i]=count
    line=line-1
print(myarray)

4
[[10. 11. 12.  1.]
 [ 9. 16. 13.  2.]
 [ 8. 15. 14.  3.]
 [ 7.  6.  5.  4.]]
>>> ================================ RESTART ================================
>>> 
6
[[16. 17. 18. 19. 20.  1.]
 [15. 30. 31. 32. 21.  2.]
 [14. 29. 36. 33. 22.  3.]
 [13. 28. 35. 34. 23.  4.]
 [12. 27. 26. 25. 24.  5.]
 [11. 10.  9.  8.  7.  6.]]
>>> ================================ RESTART ================================
>>> 
9
[[25. 26. 27. 28. 29. 30. 31. 32.  1.]
 [24. 51. 52. 53. 54. 55. 56. 33.  2.]
 [23. 50. 69. 70. 71. 72. 57. 34.  3.]
 [22. 49. 68. 79. 80. 73. 58. 35.  4.]
 [21. 48. 67. 78. 81. 74. 59. 36.  5.]
 [20. 47. 66. 77. 76. 75. 60. 37.  6.]
 [19. 46. 65. 64. 63. 62. 61. 38.  7.]
 [18. 45. 44. 43. 42. 41. 40. 39.  8.]
 [17. 16. 15. 14. 13. 12. 11. 10.  9.]]