python 输入一个自然数n然后输出一个特殊矩阵

Question

题目如图 想了好久没想出来可以怎么做，大概要用到for循环吧 求解

Answer 1

代码入下,注释很明确，不懂再问 ，请采纳，莫辜负！

# 导入 np
import numpy as np
# 输入 层数
enter = int(input())
# 计算边长
box = enter * 2 - 1
# 根据边长 生成一个 全是 1 的矩阵
a = np.ones([box, box], dtype="int64")
# 遍历 每个层
for flag in range(1, enter):
    # 遍历这个层 的 横排
    for i in range(flag, box-flag):
        # 遍历这个层 的 纵排 
        for j in range(flag, box-flag):
            # z这个位置加一
            a[i, j] += 1

print(a)