如何在 np 数组的中间插入替换值

问题描述

我有一个大小为 200x200 的零点数组。我想将中间的 16x16 零块变成 1。我该怎么做？

这是我的起始数组

cells=np.zeros((200,200),dtype = int) 形状的 (200,200) 我希望中间的 16x16 零为 1。

解决方法

这是一种更灵活的方法：

def slice_center(arr,n):
    center = arr.shape[0] // 2
    target = n // 2
    return center - target,center + target

i,j = slice_center(arr,16)
arr[i:j,i:j] = 1

,

>>> cells[92:108,92:108]
array([[1,1,1],[1,1]])

您可以计算中间部分的指数，在这种特定情况下，结果是从 92 到 108