На мой комментарий выше, вот код, который я реализовал.Кто-нибудь с предложениями, чтобы помочь мне очистить его, я буду рад обратной связи.
class Hexagon():
"""Implements a class of hexagon from a hex map which is vertically tiled.
This hexagon is able to return a list of it's neighbors. It does not care
if the neighbors are hexes which actually exist on the map or not, the map is
responsible for determining that."""
def __init__(self,grid_number):
self.name = grid_number
self.x = int(grid_number[0:2])
self.y = int(grid_number[2:4])
def neighbors(self):
ret_list = []
if self.x % 2 == 0:
temp_list = [[self.x,self.y-1],
[self.x-1,self.y], [self.x+1,self.y],
[self.x-1,self.y+1],[self.x+1,self.y+1],
[self.x,self.y+1]]
for i in temp_list:
ret_list.append(format(i[0],'02d') + format(i[1],'02d'))
elif self.x % 2 == 1:
temp_list = [[self.x,self.y-1],
[self.x-1,self.y-1],[self.x+1,self.y-1],
[self.x-1,self.y],[self.x+1,self.y],
[self.x,self.y+1]]
for i in temp_list:
ret_list.append(format(i[0],'02d') + format(i[1],'02d'))
return ret_list
def main():
hex1 = Hexagon('0201')
hex2 = Hexagon('0302')
if hex1.neighbors() == ['0200','0101','0301','0102','0302','0202']:
print("Works for even columns.")
else:
print("Failed for even columns.")
print(hex1.neighbors())
if hex2.neighbors() == ['0301','0201','0401','0202','0402','0303']:
print("Works for odd columns.")
else:
print("Failed for odd columns.")
print(hex2.neighbors())
if __name__ == '__main__':
main()