有没有Python方法来创建自定义卷积？

Question

我正在尝试创建一个极权细胞自动机的高效实现，每个细胞有三种可能的颜色，如Stephen Wolfram的书《一种新的科学》中的这张图片所示：

如果我有一个numpy数组，我是否可以创建一个新的numpy数组，方法是使用一个宽度为3的窗口在原始数组上滑动，添加三个元素，并将此总和用作规则的查找表的索引

我想我可以使用np.correlate，但这不能在规则数组中查找。也许我以后可以用一张地图来做

Answer 1

    a = self.array
    i = self.next
    window = [1, 1, 1]
    row = self.array[i - 1]
    correlated_row = np.correlate(row, window, mode="same")
    next_row = np.array([self.table[7 - total - 1] for total in correlated_row])
    a[i] = next_row
    self.next += 1