按索引或坐标访问二维数组中的元素

时间:2013-10-05 00:48:44

标签: python

设置

我正在python中编写一个操作二维方形布尔数组的类。

class Grid(object):
    def __init__(self, length):
        self.length = length
        self.grid = [[False]*length for i in range(length)]

    def coordinates(self, index):
        return (index // self.length, index % self.length)

有时在我的应用程序中,按坐标访问项目是有意义的,但有时通过索引访问项目更有意义。我还经常需要一次伪造或真实化一批物品。如果不使课堂变得非常复杂,我可以这样做:

g = Grid(8)

# access by coordinates
g.grid[4][3] = True

# access by index
coords = g.coordinates(55)
g[coords[0]][coords[1]] = True

# truthify a batch of coordinates
to_truthify = [(1, 3), (2, 3), (2, 7)]
for row, col in to_truthify:
    g.grid[row][col] = True

# falsify a batch of indices
to_falsify = [19, 22, 60]
for i in to_falsify:
    coords = g.coordinates(i)
    g.grid[coords[0]][coords[1]] = False

第一步

当然,我想在我的Grid对象中添加一些mutator方法,以便我不必直接访问内部对象并编写一堆循环:

def set_coordinate(self, row, col, value):
    self.grid[row][col] = bool(value)

def set_index(self, i, value):
    coords = self.coordinates(i)
    self.set_coordinates(coords[0], coords[1], value)

def set_coordinates(self, coordinates, value):
    for row, col in coordinates:
        self.set_coordinate(row, col, value)

def set_indices(self, indices, value):
    for i in indices:
        self.set_index(i, value)

访问者方法也很简单。我可能还想添加一些语义上有意义的别名:

def truthify_coordinate(self, row, col):
    self.set_coordinate(row, col, True)

def falsify_coordinate(self, row, col):
    self.set_coordinate(row, col, False)

def truthify_coordinates(self, coordinates):
    self.set_coordinates(coordinates, True)

... etc ...

理念

我想创建一个名为set_item的方法,其中位置可以是长度为2的可迭代,表示坐标 一个标量索引。

def set_item(self, location, value):
    try:
        location = self.coordinates(location)
    except TypeError:
        pass
    self.set_coordinates(location[0], location[1], value)

优点和缺点

这个(显然)的好处是我不需要指定位置是一对坐标还是索引,所以当我一次设置一批位置时,它们不一定都是相同的时间。例如,以下内容:

indices = [3, 5, 14, 60]
coordinates = [(1, 7), (4, 5)]
g.truthify_indices(indices)
g.truthify_coordinates(coordinates)

变为

locations = [3, 5, (1, 7), 14, (4, 5), 60]
g.truthify(locations)
在我看来,

更清晰,更易于阅读和理解。

其中一个缺点是像g.truthify((2, 3))这样的东西很难立即解密(是设置一个坐标还是两个索引?)。可能还有更多我没有想过的。

问题

实现这个想法是做pythonic的事情,还是我应该坚持明确区分索引和坐标?

1 个答案:

答案 0 :(得分:4)

我认为更多的Pythonic写作方式:

g.truthify_coordinate(row, col)

是能写的:

g[row][col] = True   # or g[row, col] = True

第二个在阅读时更容易理解,这是Python更重要的指导原则之一:code is read much more often than it is written。 (虽然我不像专家那样有资格作为Pythonic。)

那就是说,似乎你正在重新开发numpy,并且在大多数情况下重新实现一个好的工具是错误的。可能有理由不使用numpy,但应首先进行探讨。此外,如果您选择不使用numpy,它的设计将为您提供自己的方法。