python实现类似于Matlab中的magic函数

mac2022-06-30  115

参考这篇文章的代码封装了一个类似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上百实例源码以及开源项目
最新回复(0)