directional n*n grid produce 有方向性n*n grid 產生器 負號(-)代表被有edge連結

n_grid_produce.py

import random
import numpy as np
 
def proEdge(g,l):
    for t in l:
        t = t.split(',')
        i = int(t[0])
        j = int(t[1])
        if i > 0:
            if g[i-1][j]==1:
                if random.random()>0.5:
                    g[i][j]= -1
                else:
                    g[i-1][j]= -1
        if i < 3:
            if g[i+1][j]==1:
                if random.random()>0.5:
                    g[i][j]= -1
                else:
                    g[i+1][j]= -1    
        if j > 0:
            if g[i][j-1]==1:
                if random.random()>0.5:
                    g[i][j]= -1
                else:
                    g[i][j-1]= -1
        if j < 3:
            if g[i][j+1]==1:
                if random.random()>0.5:
                    g[i][j]= -1
                else:
                    g[i][j+1]= -1
return g
 
def produce(n):
    g = np.zeros((n,n),dtype=np.int)
    rannum = random.randint(4,10)
    grid = n*n
    Probability = float(rannum)/float(grid)
    l=[]
    for i in range(n):
        for j in range(n):
            if Probability>random.random():
                g[i][j]=1
                l.append(str(i)+','+str(j))
    return proEdge(g,l)
 
produce(4)