mizou - 経済統計練習帳

import numpy as np
 
def death(anx):
 
    z=[]
    for i in range(number):
        h=0
        if life[i]>60 :
            h=h+1
        if a[i]==anx :
           h=h+1
        if h==2 : 
           z.append(i)
 
    k=np.random.choice(z)
    life[k]=0
 
number=21
a=[6,5,5,4,4,4,3,3,3,3,2,2,2,2,2,1,1,1,1,1,1]
life=[100,100,100,100,100,100,100,100,100,100,100,100,100,100,100,100,100,100,100,100,100]
 
p=[]
f=[]
for i in range(15):
    p.append(-1)
    f.append(i)
 
p.append(1)    
p.append(2)
p.append(6)
p.append(7)
p.append(8)
p.append(9)
f.append(1)    
f.append(2)
f.append(6)
f.append(7)
f.append(8)
f.append(9)
 
for tr in range(5):
 
    for a1 in range(1,7):
        death(a1)
 
    px=[]  
    for i in range(number):
        h=0
        if a[i]==4:
          h=h+1
        if life[i]>50:
          h=h+1
        if h==2:
           px.append(i)
        h=0
        if a[i]==2:
          h=h+1
        if life[i]>50:
          h=h+1
        if h==2:
           px.append(i)    
 
 
    for i in range(6):
        life.append(100)
        a.append(0)
        p.append(px[i])
        sx=px[i]
        f1=f[sx]
        f.append(f1)
 
    number=number+6
 
    for i in range(number):
        a[i]=a[i]+1 
 
 
print(f)