Python 关于数独的判断

判断二维数组的每一行，每一列，每一个小方块的数字都是由0-N的数字组成的；

简单点，假如是9*9的的二维数组，那就是数组的每一行，每一列都是由0-9的数字组成，并且其中的3*3的小方块也是这样子的。

import numpy as np
array = [
  [7,8,4, 1,5,9, 3,2,6],
  [5,3,9, 6,7,2, 8,4,1],
  [6,1,2, 4,3,8, 7,5,9],

  [9,2,8, 7,1,5, 4,6,3],
  [3,5,7, 8,4,6, 1,9,2],
  [4,6,1, 9,2,3, 5,8,7],

  [8,7,6, 3,9,4, 2,1,5],
  [2,4,3, 5,6,1, 9,7,8],
  [1,9,5, 2,8,7, 6,3,4]
]

N = len(array)
for nn in range(N):
    if len(array[nn]) != N:
        print('no N*N')

n = int(N**0.5)
if n % 1 != 0:
    print('不能开平方')

temp = []
for i in range(N):
    temp.append(i+1)

array_90 = np.rot90(array)
for x in range(0, N, n):
    for y in range(0, N, n):
        box = []
        for p in range(n):
            box += array[x + p][y: y + n]
        for q in range(N):
            if sorted(array[q]) == temp and sorted(array_90[q]) == temp and sorted(box) == temp:
                print('yes')
            else:
                print('no')