Conway使用Python matplotlib的生活游戏

import numpy as np
import matplotlib.pyplot as plt

# 游戏参数
grid_size = 50  # 网格大小
num_generations = 50  # 迭代次数

# 初始化网格
grid = np.zeros((grid_size, grid_size), dtype=int)

# 设置初始细胞状态
grid[25, 25] = 1
grid[25, 26] = 1
grid[26, 25] = 1
grid[26, 27] = 1
grid[27, 26] = 1

# 迭代更新细胞状态
for generation in range(num_generations):
    new_grid = np.zeros((grid_size, grid_size), dtype=int)
    for i in range(grid_size):
        for j in range(grid_size):
            # 统计周围存活细胞数量
            num_alive_neighbors = np.sum(grid[max(0, i-1):min(i+2, grid_size), max(0, j-1):min(j+2, grid_size)]) - grid[i, j]
            if grid[i, j] == 1:
                # 存活细胞的更新规则
                if num_alive_neighbors == 2 or num_alive_neighbors == 3:
                    new_grid[i, j] = 1
            else:
                # 死亡细胞的更新规则
                if num_alive_neighbors == 3:
                    new_grid[i, j] = 1
    grid = new_grid

    # 绘制当前细胞状态
    plt.imshow(grid, cmap='binary')
    plt.title(f"Generation {generation+1}")
    plt.show()