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)

　　

测试结果

Matlab

magic(7)

Matlab