べんり - 経済統計練習帳

import numpy as np
 
def death(a1):
    q=[]
    for i in range(number):
        h=0
        if age[i]==a1:h=h+1
        if life[i]>50:h=h+1
        if h==2:q.append(i)
 
    x1=np.random.choice(q)
    life[x1]=0
 
number=21
age=[]
life=[]
for i in range(6):
    for j in range(i+1):
        age.append(6-i)
        life.append(100)
for tr in range(5):
    for i in range(6):
        death(i+1)
 
    for i in range(6):
        life.append(100)
        age.append(0)
 
    number=number+6
    for i in range(number):
        age[i]=age[i]+1
 
print(age)