python矩阵数量增加

Question

我的问题是我需要一个2维数组，该数组将值1保持在矩阵的中心，然后从那时开始其他数字。例如，当输入为3时，矩阵应该看起来像这样：

[[3, 3, 3, 3, 3],
[3, 2, 2, 2, 3],
[3, 2, 1, 2, 3],
[3, 2, 2, 2, 3],
[3, 3, 3, 3, 3]]

这是我的代码到目前为止：

import pprint

def func(number):
    d = [[0] for i in range(number + (number - 1))]
    #? Create the matrix with 0 value inside each index and determine the amount of columns

    for i in d:
        d[d.index(i)] = [number] * (number + (number - 1))
        #? Add values to the indexes
        #?              [value inside] * [length of a row]


    centre = len(d) // 2
    #? centre is the index value of the list which contains the number 1
    
    pprint.pprint(d)

func(3)

结果是：

[[3, 3, 3, 3, 3],
[3, 3, 3, 3, 3],
[3, 3, 3, 3, 3],
[3, 3, 3, 3, 3],
[3, 3, 3, 3, 3]]

我的方法是在整个桌子上填充给定的数字，然后使用桌子的其余部分工作第一个也是最后一个数组，因为它们不会更改，其价值将始终是必要的数量。

我什至知道中心是2D阵列的索引。但是，我被困在这里。我不确定如何从这里继续。

Answer 1

每个索引处的值由最大值（从行索引到中心的距离，从列索引到中心的距离）。因此，您可以做：

n = 3
dim = 2 * n - 1

result = [
    [
        max(abs(row - dim // 2), abs(col - dim // 2)) + 1
        for col in range(dim)
    ]
    for row in range(dim)
]

print(result)

这输出：

[[3, 3, 3, 3, 3],
[3, 2, 2, 2, 3],
[3, 2, 1, 2, 3],
[3, 2, 2, 2, 3],
[3, 3, 3, 3, 3]]

Answer 2

有趣的问题！

这就是我的想法。它比您的实现要短得多：

import numpy as np
mdim = (5, 5) # specify your matrix dimensions
value = 3 # specify the outer value
matrix = np.zeros(mdim) # initiate the matrix
for i, v in enumerate(range(value, 0, -1)):  # let's fill it!
    matrix[i:(mdim[0]-i), i:(mdim[1]-i)] = v
print(matrix)

输出如您所愿！可能可以对此进行更多调整。

说明：
它首先创建一个维度为 mdim 的零矩阵。
然后，它使用 0:5 行和 0:5 列的 v=3 值填充矩阵。
它依次用 1:4 行和 1:4 列的 v-1=2 值填充矩阵，直到中心得到1 的最终值。