我想修改我在正方形格子上操作的Python脚本(它是基于代理的生物学模型),以便在六边形世界中工作。
这是我在方形模型中创建和初始化2D矩阵的方法:基本上,N是晶格的大小,R给出矩阵部分的半径,我需要在算法开始时更改值:
a = np.zeros(shape=(N,N))
center = N/2
for i in xrange(N):
for j in xrange(N):
if( ( pow((i-center),2) + pow((j-center),2) ) < pow(R,2) ):
a[i,j] = 1
然后我让矩阵根据某些规则进化,最后通过创建一个pickle文件进行打印:
name = "{0}-{1}-{2}-{3}-{4}.pickle".format(R, A1, A2, B1, B2)
pickle.dump(a, open(name,"w"))
现在,我想在六角形格子上做同样的事情。我读了this有趣的StackOverflow问题,该问题清楚地表明了如何用三个坐标表示六边形网格上的位置,但是有些事情在我的知识中仍然模糊不清,即
(a)我应该如何处理Python中的三个轴,考虑到由于坐标的限制,我想要的并不等同于3D矩阵,
(b)如何策划?
至于(a),这就是我试图做的事情:
a = np.zeros(shape=(N,N,N))
for i in xrange(N/2-R, N/2+R+1):
for j in xrange(N/2-R, N/2+R+1):
for k in xrange(N/2-R, N/2+R+1):
if((abs(i)+abs(j)+abs(k))/2 <= 3*N/4+R/2):
a[i,j,k] = 1
在我看来,初始化像这样的NxNxN矩阵,然后根据坐标上的约束找到打印它的子集的方法。我正在寻找一种更简单的方法,更重要的是,要了解如何绘制算法产生的六边形格子(没有任何线索,我暂时没有尝试过任何东西)。
答案 0 :(得分:4)
我同意将六角形格子鞋拔成立方体是有问题的。我的建议是使用一般方案 - 将相邻网站表示为图。这对于pythons字典对象非常有效,并且实现&#34;轴坐标方案&#34;在您提供的其中一个链接中。这是一个创建和绘制&#34;格子的例子。使用networkx。
import networkx as nx
G = nx.Graph(directed=False)
G.add_node((0,0))
for n in xrange(4):
for (q,r) in G.nodes():
G.add_edge((q,r),(q,r-1))
G.add_edge((q,r),(q-1,r))
G.add_edge((q,r),(q-1,r+1))
G.add_edge((q,r),(q,r+1))
G.add_edge((q,r),(q+1,r-1))
G.add_edge((q,r),(q+1,r))
pos = nx.graphviz_layout(G,prog="neato")
nx.draw(G,pos,alpha=.75)
import pylab as plt
plt.axis('equal')
plt.show()
这不是最优化的实现,但它可以生成任意大的格子: