从内到外递增numpy数组中的值

import numpy as np

# 创建一个3x3的全零数组
arr = np.zeros((3, 3), dtype=int)
print("初始数组：")
print(arr)

# 定义递增的步长
step = 1

# 从内到外递增数组中的值
for i in range(min(arr.shape)//2):
    # 指定内层边界
    inner_start = i
    inner_end = arr.shape[0] - i

    # 指定外层边界
    outer_start = i
    outer_end = arr.shape[1] - i

    # 递增内层元素
    arr[inner_start:inner_end, inner_start:inner_end] += step

    # 递增外层元素
    arr[inner_start:inner_end, outer_start] += step
    arr[inner_start:inner_end, outer_end-1] += step
    arr[outer_start, inner_start+1:inner_end-1] += step
    arr[outer_end-1, inner_start+1:inner_end-1] += step

    # 增加步长
    step += 1

print("递增后的数组：")
print(arr)

递增后的数组：
[[1 1 1]
 [1 2 1]
 [1 1 1]]