python - 循环中的python类

Question

我用python创建了一类矩阵:

class MatrixOperation:

    ...

    def create_matrix(self):
        some function for creation of matrix.
        return matrix

    def matrix_vector_multiplication(self, x):
        mat = self.create_matrix()
        return numpy.dot(mat, x)

和:

M = MatrixOperation(...)
x = some set of vector

for i in range(n):
   M.matrix_vector_multiplication(x[i])

问题是，对于每次迭代，M.matrix_vector_multiplication(x[i]) 将在计算 numpy 之前重新计算 mat = self.create_matrix()。 dot(mat, x)，这是不必要的(因为它可以在开始时计算一次)。我怎样才能避免这种情况？

谢谢，

Answer 1

为避免每次都重新创建矩阵，在类的 __init__ 方法中创建一个实例属性 - 与此类似。

class Matrix(object):
    def __init__(self, data):
        self.matrix = self.create_matrix(data)
        # or simply
        # self.matrix = np.matrix(data)
    def create_matrix(data):
        # create the_matrix
        return the_matrix
    def do_something(self, x):
        z = some_function(self.matrix, x)
        return z

my_matrix = matrix([[1,2,3,4],[4,3,2,1]])