给我一个元胞自动机的Python代码

以下是一个简单的元胞自动机的 Python 代码：

import numpy as np
import matplotlib.pyplot as plt

# 初始化细胞状态
size = 100
cells = np.zeros((size, size))
cells[size//2, size//2] = 1

# 定义规则函数
def rule(cells, i, j):
    left = cells[i, (j-1)%size]
    right = cells[i, (j+1)%size]
    up = cells[(i-1)%size, j]
    down = cells[(i+1)%size, j]
    neighbors = left + right + up + down
    if neighbors == 3 or (neighbors == 2 and cells[i, j] == 1):
        return 1
    else:
        return 

# 迭代更新细胞状态
for t in range(100):
    new_cells = np.zeros((size, size))
    for i in range(size):
        for j in range(size):
            new_cells[i, j] = rule(cells, i, j)
    cells = new_cells

# 绘制结果
plt.imshow(cells, cmap='binary')
plt.show()

这个代码实现了一个简单的元胞自动机，使用了周期边界条件，规则函数是 Conway 生命游戏的规则。