Script to create a game graph for 3x3 Nim

nim-dea.py

import networkx as nx
import matplotlib.pyplot as plt
import itertools

l = list(range(4))
lists = [l, l, l]
result = []

for element in itertools.product(*lists):
    l = list(element)
    l.sort()
    result.append(tuple(l))

nodes = set(result)
edges = {}


def helper(x, y, z):
    el = []
    while x > 0:
        x = x - 1
        ele = [x, y, z]
        ele.sort()
        el.append(tuple(ele))
    return el


for a, b, c in nodes:
    edge_list = set(helper(a, b, c) + helper(b, a, c) + helper(c, a, b))
    edges[(a, b, c)] = edge_list

node_sum = 0
for n, es in edges.items():
    count = len(es)
    print(str(n) + ": " + str(count))
    node_sum += count
print(node_sum)

g = nx.DiGraph()
for n, es in edges.items():
    for e in es:
        g.add_edge(n, e)

nx.draw_spring(g, with_labels=True, scale=5, font_size=4)
plt.savefig("dea.png", dpi=300)