如何按行和列填充numpy数组

Question

我想，以填补第一和最后一行与第一和网格与最后一列4 。

输入：

grid = np.zeros((5, 5))
[[   0.   0.   0.   0.   0. ]
 [   0.   0.   0.   0.   0. ]
 [   0.   0.   0.   0.   0. ]
 [   0.   0.   0.   0.   0. ]
 [   0.   0.   0.   0.   0. ]]

输出：

[[   4.   4.   4.   4.   4. ]
 [   4.   0.   0.   0.   4. ]
 [   4.   0.   0.   0.   4. ]
 [   4.   0.   0.   0.   4. ]
 [   4.   4.   4.   4.   4. ]]

我试图通过for循环来做到这一点，但是有没有比这更好的选择了？

Answer 1

要更改第一行和最后一行的值：

grid[[0,-1],:] = 4

要更改第一列和最后一列的值：

grid[:, [0,-1]] = 4