使用Python在六边形网格中显示数据

6
我正在寻找一种方法或类,允许我在网格中显示六边形列表。理想情况下,我将能够使用某种形式的设置方法来更改单个六边形的颜色/填充/边框。
六边形使用@amitp在guide中概述的轴向坐标系存储。但是,我可以轻松地输出它们的中心作为xy坐标。
我觉得hexbinRegularPolyCollection中可能隐藏着解决方案。但前者是一个直方图方法,而后者似乎由于DPI缩放过于复杂。
那么,有没有人知道提供六角形网格的库?它不必在matplotlib中。我也很乐意使用ASCII艺术或切换到R。

1
为什么不直接使用Tkinter画布? - Arthur Vaïsse
2个回答

7

这里有一个实现方案,可以让你设置六边形单元格的颜色,并且还可以创建自定义边框颜色。

这是全手工制作的,只用了1小时参考你的网站。你可能需要根据自己的需求进行适应,但似乎可以正常工作。

from tkinter import *

class HexaCanvas(Canvas):
    """ A canvas that provides a create-hexagone method """
    def __init__(self, master, *args, **kwargs):
        Canvas.__init__(self, master, *args, **kwargs)
    
        self.hexaSize = 20
    
    def setHexaSize(self, number):
        self.hexaSize = number
    
    
    def create_hexagone(self, x, y, color = "black", fill="blue", color1=None, color2=None, color3=None, color4=None, color5=None, color6=None):
        """ 
        Compute coordinates of 6 points relative to a center position.
        Point are numbered following this schema :
    
        Points in euclidiean grid:  
                    6
                    
                5       1
                    .
                4       2
            
                    3
    
        Each color is applied to the side that link the vertex with same number to its following.
        Ex : color 1 is applied on side (vertex1, vertex2)
    
        Take care that tkinter ordinate axes is inverted to the standard euclidian ones.
        Point on the screen will be horizontally mirrored.
        Displayed points:
    
                    3
              color3/      \color2      
                4       2
            color4|     |color1
                5       1
              color6\      /color6
                    6
        
        """
        size = self.hexaSize
        Δx = (size**2 - (size/2)**2)**0.5
    
        point1 = (x+Δx, y+size/2)
        point2 = (x+Δx, y-size/2)
        point3 = (x   , y-size  )
        point4 = (x-Δx, y-size/2)
        point5 = (x-Δx, y+size/2)
        point6 = (x   , y+size  )
    
        #this setting allow to specify a different color for each side.
        if color1 == None:
            color1 = color
        if color2 == None:
            color2 = color
        if color3 == None:
            color3 = color
        if color4 == None:
            color4 = color
        if color5 == None:
            color5 = color
        if color6 == None:
            color6 = color
    
        self.create_line(point1, point2, fill=color1, width=2)
        self.create_line(point2, point3, fill=color2, width=2)
        self.create_line(point3, point4, fill=color3, width=2)
        self.create_line(point4, point5, fill=color4, width=2)
        self.create_line(point5, point6, fill=color5, width=2)
        self.create_line(point6, point1, fill=color6, width=2)
    
        if fill != None:
            self.create_polygon(point1, point2, point3, point4, point5, point6, fill=fill)
    
class HexagonalGrid(HexaCanvas):
    """ A grid whose each cell is hexagonal """
    def __init__(self, master, scale, grid_width, grid_height, *args, **kwargs):
    
        Δx     = (scale**2 - (scale/2.0)**2)**0.5
        width  = 2 * Δx * grid_width + Δx
        height = 1.5 * scale * grid_height + 0.5 * scale

        HexaCanvas.__init__(self, master, background='white', width=width, height=height, *args, **kwargs)
        self.setHexaSize(scale)
    
    def setCell(self, xCell, yCell, *args, **kwargs ):
        """ Create a content in the cell of coordinates x and y. Could specify options throught keywords : color, fill, color1, color2, color3, color4; color5, color6"""
    
        #compute pixel coordinate of the center of the cell:
        size = self.hexaSize
        Δx = (size**2 - (size/2)**2)**0.5
    
        pix_x = Δx + 2*Δx*xCell
        if yCell%2 ==1 :
            pix_x += Δx
    
        # Add 5 to avoid clipping on top row of hexes
        pix_y = size + yCell*1.5*size + 5
    
        self.create_hexagone(pix_x, pix_y, *args, **kwargs)
    
    
    
if __name__ == "__main__":
    tk = Tk()

    grid = HexagonalGrid(tk, scale = 50, grid_width=4, grid_height=4)
    grid.grid(row=0, column=0, padx=5, pady=5)

    def correct_quit(tk):
        tk.destroy()
        tk.quit()

    quit = Button(tk, text = "Quit", command = lambda :correct_quit(tk))
    quit.grid(row=1, column=0)

    grid.setCell(0,0, fill='blue')
    grid.setCell(1,0, fill='red')
    grid.setCell(0,1, fill='green')
    grid.setCell(1,1, fill='yellow')
    grid.setCell(2,0, fill='cyan')
    grid.setCell(0,2, fill='teal')
    grid.setCell(2,1, fill='silver')
    grid.setCell(1,2, fill='white')
    grid.setCell(2,2, fill='gray')

    tk.mainloop()

我尽力对我的代码进行了适当的注释。如果您有任何不清楚的地方,请不要犹豫,请求解释。

祝你好运 Arthur Vaisse。

NB:脚本在Python 3上运行。绘图有些粗糙。可以像https://mail.python.org/pipermail/tkinter-discuss/2009-April/001904.html中建议的那样添加抗锯齿来改善Tk画布。


在Win7中,3.4.2版本的tk 8.6中,tk.quit退出tcl但不关闭tk窗口。而tk.destroy则可以关闭窗口。 - Terry Jan Reedy
在我的Linux上它可以正常工作。我知道为了正确退出它,你必须调用tk.destroytk.quit方法。但感谢你在这里指出这个错误。我会编辑代码的。 - Arthur Vaïsse

1
你可以使用我的 hexalattice Python 模块来创建和绘制六边形网格:
from hexalattice.hexalattice import *
hex_centers, _ = create_hex_grid(nx=5,
                                 ny=5,
                                 do_plot=True)
                                 
plt.show()    # import matplotlib.pyplot as plt

生成这个图表:输入图像描述

查看此答案以获取更多示例: 此答案


网页内容由stack overflow 提供, 点击上面的
可以查看英文原文,
原文链接