matlab的magic函数,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

原文：http://www.cnblogs.com/ssooking/p/7502244.html