7-26 输出一个 n 阶的方阵

读入 1 个正整数 n(3≤n<9)和 1 个整数 digit（2<=digit<=9）, 请输出一个 n 阶的方阵, 该矩阵所有边上的元素都是 digit, 其它元素都是 digit-1.

输入格式:

在一行中输入n和digt

输出格式:

输出n 阶的方阵

输入样例:

在这里给出一组输入。例如：

4   2

输出样例:

在这里给出相应的输出。例如：

2 2 2 2 
2 1 1 2 
2 1 1 2 
2 2 2 2 

参考代码

# 获取用户输入的两个整数，并使用空格分隔开
a, b = map(int, input().split())

# 使用循环逐行打印图案
for line in range(1, a+1):
    if line == 1:
        # 如果是第一行，打印 b 的值 a 次，以空格分隔
        print("{} ".format(b) * a)
    elif line == a:
        # 如果是最后一行，打印 b 的值 a 次，以空格分隔
        print("{} ".format(b) * a)
    else:
        # 其他行，打印 b 的值一次，然后打印 b-1 的值 (a-2) 次，再打印 b 的值一次
        print("{} ".format(b) + "{} ".format(b-1) * (a-2) + "{} ".format(b))