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

Posted 2020-10-06

tags:

篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了python实现类似于Matlab中的magic函数相关的知识，希望对你有一定的参考价值。

参考这篇文章的代码封装了一个类似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)

测试结果

技术分享