９ます - 経済統計練習帳

import numpy as np
 
def makedir(h1,k1):
 
    dir=[]
    if k1=="王": 
       dir.append([1,0])
       dir.append([-1,0])
       dir.append([0,1])
       dir.append([0,-1])
       dir.append([1,1])
       dir.append([1,-1])
       dir.append([-1,1])
       dir.append([-1,-1])
 
 
    h=0
    if k1=="歩":
       h=h+1
    if h1=="先手":
       h=h+1
    if h==2:
       dir.append([0,1])
 
    h=0
    if k1=="歩":
       h=h+1
    if h1=="後手":
       h=h+1
    if h==2:
       dir.append([0,-1])   
    return dir
 
 
def select_koma(h1):
 
    qx=[]
    for i in range(6):
        if hand[i]==h1:
           qx.append(i)
    q1=np.random.choice(qx)
    return q1
 
ban=[[0,0],[1,1],[2,1],[3,1],[1,2],[2,2],[3,2],[1,3],[2,3],[3,3]]
koma=["王","歩","歩","王","歩","歩"]
hand=["先手","先手","先手","後手","後手","後手"]
 
q=[1,2,3,7,8,9,"先手"]
 
k1=select_koma("先手")
 
print(k1)
 
dir=makedir("先手",koma[k1])
 
print(dir)
 
qq=q
c=np.random.choice(range(len(dir)))
d1=dir[c]
 
print(d1)
 
m=q[k1]
x1=ban[m][0]+d1[0]
y1=ban[m][1]+d1[1]
sxs=-1
if [x1,y1] in ban:
   sxs=ban.index([x1,y1])
if sxs>0:
   qq[6]="後手"
   qq[k1]=sxs
 
print(qq)