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