anx.py

from itertools import product
import numpy as np

def enumerate_ANX(oxi_states, max_ratio=8):
    ret = []
    for anx in product(range(1, max_ratio + 1), repeat=len(oxi_states)):
        if np.sum(np.array(oxi_states) * np.array(anx)) == 0:
            ret.append(anx)
    return ret