参考这篇文章的代码封装了一个类似Matlab中的magic函数,用来生成魔方矩阵。
#!/usr/bin/env python
# -*- coding: utf-8 -*-
import numpy as np
def magic(n):
row,col=0,n//2
magic=[]
for i in range(n):
magic.append([0]*n)
magic[row][col]=1
for i in range(2,n*n+1):
r,l=(row-1+n)%n,(col+1)%n
if(magic[r][l]==0):
row,col=r,l
else:
row=(row+1)%n
magic[row][col]=i
marray = np.array(magic)
return marray
if __name__ == '__main__':
y = magic(7)
print y
print type(y)
测试结果
Matlab
magic(7)
Matlab
转载于:https://www.cnblogs.com/ssooking/p/7502244.html
相关资源:JAVA上百实例源码以及开源项目