停止动画 - 康威的生命游戏

时间:2017-08-12 18:24:38

标签: python numpy animation matplotlib

以下是我在两种合作细胞之间制作的康威生命游戏的代码。 让我们说10代后,我想停止动画。 我尝试使用"帧"动画,但动画不会停止。 如何在10个生成后停止它? 这是代码:

import numpy as np
import matplotlib.pyplot as plt 
import matplotlib.animation as animation
import random

N = 100
OnTypeOne = 10
OnTypeTwo= -10
OFF = 0
vals = [OnTypeOne , OnTypeTwo, OFF]


# populate grid with random on/off - more off than on
grid = np.random.choice(vals, N*N, p=[0.2,0.2, 0.6]).reshape(N, N)

def update(data):
  global grid
  x=0
  # copy grid since we require 8 neighbors for calculation
  # and we go line by line 
  newGrid = grid.copy()
  for i in range(N):
    for j in range(N):
      # compute 8-neghbor sum 
      # using toroidal boundary conditions - x and y wrap around 
      # so that the simulaton takes place on a toroidal surface.
      total = (abs (grid[i, (j-1)%N])+ abs (grid[i, (j+1)%N]) + 
           abs (grid[(i-1)%N, j]) + abs (grid[(i+1)%N, j]) + 
           abs (grid[(i-1)%N, (j-1)%N]) + abs (grid[(i-1)%N, (j+1)%N]) + 
           abs (grid[(i+1)%N, (j-1)%N]) + abs (grid[(i+1)%N, (j+1)%N]))/10
      # apply Conway's rules
      if grid[i, j]  == OnTypeOne:
        if (total < 2) or (total > 3):
          newGrid[i, j] = OFF
        if (total==3):
           newGrid[i, j] = OnTypeOne
      if  grid[i, j]  == OnTypeTwo:
          if (total < 2) or (total > 3):
              newGrid[i, j] = OFF
          if (total==3):
              newGrid[i, j] = OnTypeTwo
      if grid[i, j]  == OFF:
          if total==3:
              x=random.random()
              if x<=0.5:
                  newGrid[i,j]=OnTypeOne
              elif x>0.5:
                  newGrid[i,j]=OnTypeTwo
          if total!=3:
              newGrid[i,j]=OFF




  # update data
  mat.set_data(newGrid)
  grid = newGrid
  return [mat]

# set up animation
fig, ax = plt.subplots()
mat = ax.matshow(grid)

ani = animation.FuncAnimation(fig, update, frames=10, 
interval=50,save_count=50
                         , blit=True )

 plt.show()

1 个答案:

答案 0 :(得分:1)

您需要添加参数repeat=False以告知动画不要在10帧后重新启动,

ani = animation.FuncAnimation(fig, update, frames=10, 
                              interval=50, save_count=50, blit=True, repeat=False )