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 age=[] life=[] number=21 for i in range(6): for j in range(i+1): age.append(6-i) life.append(100) for a1 in range(6): death(a1+1) print(life)