将int数组转换为矩阵类以完成运算符重载

class Matrix:
    def __init__(self, array):
        self.matrix = array

    def __add__(self, other):
        result = []
        for i in range(len(self.matrix)):
            row = []
            for j in range(len(self.matrix[0])):
                row.append(self.matrix[i][j] + other.matrix[i][j])
            result.append(row)
        return Matrix(result)

    def __sub__(self, other):
        result = []
        for i in range(len(self.matrix)):
            row = []
            for j in range(len(self.matrix[0])):
                row.append(self.matrix[i][j] - other.matrix[i][j])
            result.append(row)
        return Matrix(result)

    def __mul__(self, other):
        result = []
        for i in range(len(self.matrix)):
            row = []
            for j in range(len(self.matrix[0])):
                value = 0
                for k in range(len(self.matrix[0])):
                    value += self.matrix[i][k] * other.matrix[k][j]
                row.append(value)
            result.append(row)
        return Matrix(result)

    def transpose(self):
        result = []
        for i in range(len(self.matrix[0])):
            row = []
            for j in range(len(self.matrix)):
                row.append(self.matrix[j][i])
            result.append(row)
        return Matrix(result)

    def get_row_count(self):
        return len(self.matrix)

    def get_col_count(self):
        return len(self.matrix[0])

# 示例使用
array1 = [[1, 2], [3, 4]]
array2 = [[5, 6], [7, 8]]
matrix1 = Matrix(array1)
matrix2 = Matrix(array2)

# 矩阵相加
result_add = matrix1 + matrix2
print(result_add.matrix)

# 矩阵相减
result_sub = matrix1 - matrix2
print(result_sub.matrix)

# 矩阵相乘
result_mul = matrix1 * matrix2
print(result_mul.matrix)

# 矩阵转置
result_transpose = matrix1.transpose()
print(result_transpose.matrix)

# 获取矩阵的行数和列数
row_count = matrix1.get_row_count()
col_count = matrix1.get_col_count()
print(row_count, col_count)