ato - 経済統計練習帳

def select(m1):
 
    kon=[]
    for u in range(6):
        if hand[u]==m1:
           kon.append(u)
 
    return kon 
 
def zero():
 
    sig=[1,2,3,4,5,6,7,8,9]
 
    for u in pos:
        sig.remove(u)
 
    return sig
 
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
 
mode="先手"
koma=["王","歩","歩","王","歩","歩"]
hand=["先手","先手","先手","後手","後手","後手"]
pos=[2,1,3,8,7,9]
ban=[[0,0],[1,1],[2,1],[3,1],[1,2],[2,2],[3,2],[1,3],[2,3],[3,3]]
 
 
kk=select(mode)
 
print(kk)
 
u=kk[0]
 
dir=makedir(hand[u],koma[u])
 
d=dir[0]
 
print(d)
 
p=pos[u]
 
b=[ban[p][0]+d[0],ban[p][1]+d[1]]
 
p1=-1
if b in ban:
   p1=ban.index(b)
 
px=pos
 
px[u]=p1
 
print(px)
 
mm="後手"
if mode=="後手":
    mm="先手" 
　
for i in range(6):
    h=0
    if px[i]==p1:
       h=h+1
    if hand[i]==mm:
       h=h+1
    if h==2:
       hand[i]=mode
 
 
de=0
for i in range(6):
    h=0
    if px[i]==p1:
       h=h+1
    if hand[i]==mode:
       h=h+1
    if h==2:
       de=de+1
 
print(de)
 
print(zero())