Python实现螺旋矩阵的填充算法示例

#coding=utf-8import numpy'''''Author: afantyDate:  2016/6/23'''def helixMatrix(n):  '''''实现n维螺旋矩阵的填充  :param n:维数  :return:螺旋矩阵  '''  if not isinstance(n, int) or n <= 0:    raise ValueError('请输入合适的维数')  matrix = numpy.zeros((n, n))  left_top = 0  right_buttom = n - 1  number = 1  while left_top < right_buttom:    # 向右移动，横坐标不变，纵坐标+1，number+1    i = left_top    while i < right_buttom:      matrix[left_top][i] = number      i += 1      number += 1    # while    # 向下移动，纵坐标不变，横坐标+1，number+1    i = left_top    while i < right_buttom:      matrix[i][right_buttom] = number      i += 1      number += 1    #while    # 向左移动，横坐标不变，纵坐标-1，number+1    i = right_buttom    while i > left_top:      matrix[right_buttom][i] = number      i -= 1      number += 1    # while    # 向上移动，纵坐标不变，横坐标-1，number+1    i = right_buttom    while i > left_top:      matrix[i][left_top] = number      i -= 1      number += 1    # while    left_top += 1    right_buttom -= 1  # while  if n % 2 != 0:    matrix[n / 2][n / 2] = n * n  return matrix# endprint("武林网测试结果：")print helixMatrix(5)