Python 两个矩阵相加

# 两个n行n列的矩阵相加，实现其对应位置的数据相加，返回一个新矩阵。

import random


def gen_matrix(n, a=0, b=0):
    """
    生成一个 n × n 的矩阵，矩阵中的每个元素都是在 a 和 b 之间的随机整数。

    参数:
    n (int): 矩阵的行数和列数。
    a (int): 随机数范围的下限，默认为 0。
    b (int): 随机数范围的上限，默认为 0。

    返回:
    list: 一个 n × n 的矩阵，其中每个元素都是在 a 和 b 之间的随机整数。
    """
    # 初始化一个空矩阵
    arr = []
    for i in range(n):
        arr.append([])
        # 为矩阵的每个元素生成随机数
        for j in range(n):
            arr[i].append(random.randint(a, b))
    return arr


def get_sum():
    """
    生成两个随机数 3×3 矩阵，并计算它们的和，然后打印结果。
    """
    # 定义矩阵的大小和随机数范围
    n = 3
    a = 1
    b = 100
    m1 = gen_matrix(n, a, b)  # 生成第一个随机矩阵
    m2 = gen_matrix(n, a, b)  # 生成第二个随机矩阵
    result = gen_matrix(n)  # 初始化存储结果的矩阵

    # 计算两个矩阵的和
    for i in range(n):
        for j in range(n):
            result[i][j] = m1[i][j] + m2[i][j]

    # 打印两个随机矩阵和它们的和
    print(f'两个随机数{n}*{n}矩阵:')
    [print(m1[i]) for i in range(n)]
    print('')
    [print(m2[i]) for i in range(n)]
    print('')
    print('相加结果:')
    [print(result[i]) for i in range(n)]


get_sum()