创建一个矩阵，其中每个元素等于其行和列索引的最小值

Question

我想创建一个矩阵C，其中每个元素等于其对应的行和列索引的最小值。 例如：与第一行和第二列对应的元素的值应为1，与第八行和第三列对应的元素的值应为3，依此类推。

我编写了以下代码，这些代码可以使我得到想要的东西。 运行以下代码：

from numpy import empty

C = empty(shape=(32,32))

for j in range(1,33):
    for i in range(1,33):
        minimum = min(i,j)
        C[i-1][j-1] = minimum

print(C)

结果是

[[  1.   1.   1. ...,   1.   1.   1.]
 [  1.   2.   2. ...,   2.   2.   2.]
 [  1.   2.   3. ...,   3.   3.   3.]
 ..., 
 [  1.   2.   3. ...,  30.  30.  30.]
 [  1.   2.   3. ...,  30.  31.  31.]
 [  1.   2.   3. ...,  30.  31.  32.]]

问题：这是最有效的方法吗？ 如果不; 如何改进这种方法？

Answer 1

选项1
np.mgrid

np.mgrid[1:33, 1:33].min(axis=0)

array([[ 1,  1,  1, ...,  1,  1,  1],
       [ 1,  2,  2, ...,  2,  2,  2],
       [ 1,  2,  3, ...,  3,  3,  3],
       ...,
       [ 1,  2,  3, ..., 30, 30, 30],
       [ 1,  2,  3, ..., 30, 31, 31],
       [ 1,  2,  3, ..., 30, 31, 32]])

---

选项2
np.indices

(np.indices((32, 32)) + 1).min(axis=0)

array([[ 1,  1,  1, ...,  1,  1,  1],
       [ 1,  2,  2, ...,  2,  2,  2],
       [ 1,  2,  3, ...,  3,  3,  3],
       ...,
       [ 1,  2,  3, ..., 30, 30, 30],
       [ 1,  2,  3, ..., 30, 31, 31],
       [ 1,  2,  3, ..., 30, 31, 32]])

Answer 2

另一种方法是执行上三角矩阵的累加总和，该矩阵沿每一列包含全1：

In [16]: np.cumsum(np.triu(np.ones((32,32))), axis=0)
Out[16]:
array([[  1.,   1.,   1., ...,   1.,   1.,   1.],
       [  1.,   2.,   2., ...,   2.,   2.,   2.],
       [  1.,   2.,   3., ...,   3.,   3.,   3.],
       ...,
       [  1.,   2.,   3., ...,  30.,  30.,  30.],
       [  1.,   2.,   3., ...,  30.,  31.,  31.],
       [  1.,   2.,   3., ...,  30.,  31.,  32.]])

显然不如mgrid方法有效，但我认为这是一种不错的选择。