numpy 如何实现一个正方形边缘区域置为 0

V2EX = way to explore

V2EX 是一个关于分享和探索的地方

现在注册

已注册用户请登录

V2EX 提问指南

这是一个创建于 2577 天前的主题，其中的信息可能已经有所发展或是发生改变。

一张这样的 numpy 矩阵，圆的直径等于正方形变长，我想把圆外的区域都置为 0 或者一个边界值，python 有什么有效的方法吗？
谢谢各位！

NumPy

方形

img

Python

1 条回复 • 2017-11-03 18:30:37 +08:00

Hzzone

2017-11-03 18:30:37 +08:00

更新一下解决方法：
```
width, height = pixel_array.shape
y, x = np.ogrid[0:width,0:height]
centerx, centery = (width/2, height/2)
# print(x, y)
mask = ((y - centery)**2 + (x - centerx)**2) > (width/2)**2
print(mask.shape)
pixel_array[mask] = YOUR_BOUNDARY
plt.imshow(pixel_array)
plt.show()
plt.hist(pixel_array.flatten(), bins=80, color='c')
plt.show()
```