Python实现康威生命游戏的问题

import numpy as np

cells = np.array([[0, 1, 0],
                  [0, 1, 0],
                  [0, 1, 0]])

def get_next_state(cells):
    # 获取细胞的周围8个细胞的状态
    neighborhood = np.zeros((3, 3), dtype=int)
    neighborhood[1, 1] = cells[1, 1]
    neighborhood[0, 1] = cells[0, 1]
    neighborhood[2, 1] = cells[2, 1]
    neighborhood[1, 0] = cells[1, 0]
    neighborhood[1, 2] = cells[1, 2]

    # 计算周围活细胞的数量
    num_neighbors = np.sum(neighborhood)

    # 根据规则计算细胞的下一个状态
    if num_neighbors == 3 or (num_neighbors == 2 and cells[1, 1] == 1):
        return 1
    else:
        return 0

while True:
    # 计算每个细胞的下一个状态
    next_cells = np.zeros_like(cells)
    for i in range(3):
        for j in range(3):
            next_cells[i, j] = get_next_state(cells[i-1:i+2, j-1:j+2])

    # 更新细胞的状态
    cells = next_cells

    # 输出当前状态
    print(cells)

    # 如果游戏结束，则退出循环
    if np.all(cells == 0):
        break