python - 在列号是操作数之一的情况下进行操作？

Question

假设我有 (5,5) 数组。如果数组全为零，那么如果我这样做，我该如何进行操作，以使列号是操作数 fe 之一：

 a + Col-num?

那么结果的第一列将全为 1，第二列全为 2，...3，4，全为 5

那么第二个幽灵操作数是列号？

标签： pythonnumpy

Answer 1

我认为这会比发布的答案快一点：

def f(c):
    col, arr = c
    return arr + (col + 1)
a = np.zeros((5, 5))
columnAdded = np.apply_along_axis(f, 1, enumerate(a.T)).T
columnAdded
>>> array([[1., 2., 3., 4., 5.],
       [1., 2., 3., 4., 5.],
       [1., 2., 3., 4., 5.],
       [1., 2., 3., 4., 5.],
       [1., 2., 3., 4., 5.]])

numpy按要求完成