您好, 欢迎来到 !    登录 | 注册 | | 设为首页 | 收藏本站

Python中的多维数组

Python中的多维数组

您可以使用嵌套列表创建它:

matrix = [[a,b],[c,d],[e,f]]

如果必须动态的话,它会更加复杂,为什么不自己编写一个小类呢?

class Matrix(object):
    def __init__(self, rows, columns, default=0):
        self.m = []
        for i in range(rows):
            self.m.append([default for j in range(columns)])

    def __getitem__(self, index):
        return self.m[index]

可以这样使用:

m = Matrix(10,5)
m[3][6] = 7
print m[3][6] // -> 7

我敢肯定,可以更有效地实施它。:)

如果您需要多维数组,则可以创建一个数组并计算偏移量,也可以在数组中的数组中使用数组,这对于内存来说可能是非常糟糕的。(不过可能会更快…)我实现了第一个想法,例如:

class Matrix(object):
    def __init__(self, *dims):
        self._shortcuts = [i for i in self._create_shortcuts(dims)]
        self._li = [None] * (self._shortcuts.pop())
        self._shortcuts.reverse()

    def _create_shortcuts(self, dims):
        dimList = list(dims)
        dimList.reverse()
        number = 1
        yield 1
        for i in dimList:
            number *= i
            yield number

    def _flat_index(self, index):
        if len(index) != len(self._shortcuts):
            raise TypeError()

        flatIndex = 0
        for i, num in enumerate(index):
            flatIndex += num * self._shortcuts[i]
        return flatIndex

    def __getitem__(self, index):
        return self._li[self._flat_index(index)]

    def __setitem__(self, index, value):
        self._li[self._flat_index(index)] = value

可以这样使用:

m = Matrix(4,5,2,6)
m[2,3,1,3] = 'x'
m[2,3,1,3] // -> 'x'
python 2022/1/1 18:16:22 有571人围观

撰写回答


你尚未登录,登录后可以

和开发者交流问题的细节

关注并接收问题和回答的更新提醒

参与内容的编辑和改进,让解决方法与时俱进

请先登录

推荐问题


联系我
置顶