如何按行和列填充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

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

WaveFormat

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

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