Игра в пятнадцать

import copy
import random
#import numpy as np
 
data=[3,1,6,7,0,5,4,2,8]
H=3
W=3
 
print('начальная позиция')    
for i in range(H):
    for j in range(W):
        print ("%i"%data[i*(W)+j],end=" ")
    print()
print( )
 
    
def next_state(data_):
    #найти позицию 0(i,j)
    index = data_.index(0)
    i = index//H
    j = index%H
    #print(i,'   ',j)
    #найти позиции клеток которые могут переместиться в 0(пустую)
    variants = 4
    possible=[1,1,1,1] #список содержит позиции возможных ячеек(left,right, top,bottom)
    left_x = i
    left_y = j-1
 
    right_x = i
    right_y = j+1
 
    top_x = i-1
    top_y = j
 
    bottom_x = i+1
    bottom_y = j
 
    
 
    #отсеить клетки, которые невозможно реализовать
    if left_y == -1:
        variants -= 1
        possible[0]=0
    if right_y == 3:
        variants -= 1
        possible[1]=0
    if top_x == -1:
        variants -= 1
        possible[2]=0
    if bottom_x == +3:
        variants -= 1
        possible[3]=0
    #print(variants)
    #print(possible)
    #сгенерировать позиции, которые взможны из текущей
    next_s=[]
    if possible[0] == 1:
        # преобразовываем столбец и строку в одномерный индекс
        index_2 = left_x * (W) + left_y
        # делаем копию начального массива
        data_2 = copy.deepcopy(data_)
        #меняем местами две ячейки
        data_2[index_2], data_2[index]=data_2[index],data_2[index_2]
        #data_2 = np.array(data_2)
        next_s.append(data_2)
    
 
    if possible[1] == 1:
        # преобразовываем столбец и строку в одномерный индекс
        index_2 = right_x * (W) + right_y
        # делаем копию начального массива
        data_3 = copy.deepcopy(data_)
        #меняем местами две ячейки
        data_3[index_2], data_3[index]=data_3[index],data_3[index_2]
        #data_3 = np.array(data_3)
        next_s.append(data_3)
    
 
    #вывести эти позиции на экран
    if possible[2] == 1:
        # преобразовываем столбец и строку в одномерный индекс
        index_2 = top_x * (W) + top_y
        # делаем копию начального массива
        data_4 = copy.deepcopy(data_)
        #меняем местами две ячейки
        data_4[index_2], data_4[index]=data_4[index],data_4[index_2]
        #data_4 = np.array(data_4)
        next_s.append(data_4)
    
 
    #вывести эти позиции на экран
    if possible[3] == 1:
        # преобразовываем столбец и строку в одномерный индекс
        index_2 = bottom_x * (W) + bottom_y
        # делаем копию начального массива
        data_5 = copy.deepcopy(data_)
        #меняем местами две ячейки
        data_5[index_2], data_5[index]=data_5[index],data_5[index_2]
        #data_5 = np.array(data_5)
        next_s.append(data_5)
    
        
   #next_s = np.vstack((data_2,data_3,data_4,data_5))
    return variants, next_s
 
 
   
#вывести эти позиции на экран
V, s = next_state(data)
for v in range(V):
    for i in range(H):
        for j in range(W):
            print ("%i"%s[v][i*(W)+j],end=" ")
        print()
    print( )