创建一个 class 接受矩阵进行实例化

Create a class that takes a matrix for instantiation

我被要求创建一个 class,它采用 m x n 数组进行实例化。然后给 class 一个对矩阵进行任何操作的方法,假设将其对角线项加倍并 returns 它。

从这个简单的框架开始:

class Matrix:

    def __init__(self, matrix):
        self.matrix = matrix

    def double_diagnonal_entries(self):
        # do calcs
        return self.matrix

请注意,如果您需要实现一些基本的矩阵运算,例如加法,您可能会考虑运算符重载,例如:

def __add__(self, another_matrix):
    # do the math
    return sum_matrix