矩阵中的幸运数Python代码怎么写

矩阵中的幸运数是指在一个矩阵中找到一个数字，它的行、列以及对角线上的数值之和都相等。以下是使用Python编写的一个简单函数，用于寻找给定矩阵中的幸运数：

def find_lucky_number(matrix):
    # 检查矩阵是否为空或只有一行一列
    if len(matrix) == 0 or (len(matrix) == 1 and len(matrix[0]) == 1):
        return None

    # 初始化幸运数变量
    lucky_num = sum(matrix[0])
    
    # 使用两层循环遍历矩阵
    for i in range(1, len(matrix)):
        for j in range(len(matrix[i])):
            # 更新幸运数，包括当前元素值，以及左上到右下的对角线和左下到右上的对角线
            lucky_num -= matrix[i][j]
            if i + j < len(matrix):
                lucky_num += matrix[i][j]  # 对角线1
            if i - j >= 0:
                lucky_num += matrix[i][j]  # 对角线2
                
    # 如果所有行、列和对角线和都相等，则返回这个数，否则返回None
    return lucky_num if lucky_num * 4 == sum(sum(row) for row in matrix) else None

# 示例矩阵
matrix_example = [
    [2, 7, 4],
    [9, 5, 1],
    [6, 3, 8]
]

# 找到幸运数
lucky_num = find_lucky_number(matrix_example)
if lucky_num is not None:
    print("矩阵中的幸运数是:", lucky_num)
else:
    print("找不到幸运数.")