Python中的多维数组

一尘不染

Python中的多维数组

java

我有一个要转换为Python的Java小问题。因此，我需要一个多维数组。在Java中，它看起来像：

double dArray[][][] = new double[x.length()+1][y.length()+1][x.length()+y.length()+3];
dArray[0][0][0] = 0;
dArray[0][0][1] = POSITIVE_INFINITY;

其他值将在bei循环中创建并写入数组。

如何实例化数组？

PS：不涉及矩阵乘法…

阅读 651

2020-12-03

共1个答案

一尘不染

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

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'

2020-12-03