Conway的生命游戏使用Python matplotlib

时间:2017-01-21 23:11:17

标签: python numpy animation matplotlib

我刚开始学习Python,所以我们必须做这个功课,编写一个程序来运行使用matplotlib的Conway的生命游戏。我的教授已完成部分代码,我们必须填写TODO评论下的代码。所以我写了它,我不知道它为什么没有运行动画。我做错了什么?非常感谢你们!该计划中的评论很长,因为它们是一种指示。

# life.py - Once complete, this program will play the game of
#           life.

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

# The update function is needed for the animation
#   to work properly. It has four parameters:
# frameNum - this is handled by the animation, don't change this.
# img - the plot that is passed and changed, don't change this.
# world - the two-D array that represents the world for the
#         game of life. This will be updated to the next gen.
# N - the size of the world (which is square).

def update( frameNum, img, world, N ) :

    newWorld = world.copy( )

    ## TODO - Write code here to update the cells in newWorld
    ##        for the next generation of life. Remember to
    ##        use the toroidal model. Rather than making
    ##        special cases for row/column 0 and N-1, just
    ##        use the % operator.

    for i in range (N):
        for j in range (N):
            total= (world[(i-1)%N][(j-1)%N]+world[(i-1)%N][j]+world[(i-1)%N][(j+1)%N]+
                   world[i][(j-1)%N]+world[i][(j+1)%N]+ world[(i+1)%N][(j-1)%N]+
                   world[(i+1)%N][j]+ world[(i+1)%N][(j+1)%N])/255

            if world[i][j]== 255:

                if total>3 or total<2:
                    newWorld[i][j]== 0
                elif total==3:
                    newWorld[i][j]== 255

    img.set_data( newWorld )
    world[:] = newWorld[:]
    return img

N = 50
SPEED = 100
PROB_LIFE = 40


## TODO - Write code here to create the initial population
##   of the world. I recommend creating an N x N array that
##   is initially filled with random numbers from 0 to 99.
##   Then use the PROB_LIFE to change the entries to be
##   either alive (255) or dead (0).
world= np.random.choice([0,255], N*N, p=[1-((PROB_LIFE)/100),(PROB_LIFE)/100]).reshape(N,N)

fig, ax = plt.subplots( )
img = ax.imshow( world, interpolation='nearest' )
ani = animation.FuncAnimation( fig, update, fargs = ( img, world, N ),
                               frames = 10, interval = SPEED,
                               save_count = 50 )
plt.show( )

# This is the end of the program. You should not need
#   anything after this point.

1 个答案:

答案 0 :(得分:3)

您必须使用set代替 .... (S_ID, S_LAST, S_FIRST, S_MI, S_ADDRESS, S_CITY, S_STATE, S_ZIP, S_PHONE, S_CLASS, @S_DOB, S_PIN, F_ID, @DATE_ENROLLED); SET S_DOB = STR_TO_DATE(@S_DOB, '%m/%d/%Y'), DATE_ENROLLED = STR_TO_DATE(@DATE_ENROLLED, '%m/%d/%Y');

中指定新值
=

现在你会看到动画。

但你也错误地使用== - 错误的缩进 - 正确

newWorld[i][j] = 0   # need `=` instead of `==

newWorld[i][j] = 255 # need `=` instead of `==

或更具可读性

elif