如何为Numpy数组创建切片对象?

2024-06-16 14:00:31 发布

您现在位置:Python中文网/ 问答频道 /正文

我试图找到一个简洁的解决方案,但我正在以相同的方式切片几个相同形状的二维数组。我通过定义一个包含“x,y”中心的列表来尽可能地整理它,例如cpix = [161, 134]我想做的是不必像这样将切片写三次:

a1 = array1[cpix[1]-50:cpix[1]+50, cpix[0]-50:cpix[0]+50] 
a2 = array2[cpix[1]-50:cpix[1]+50, cpix[0]-50:cpix[0]+50] 
a3 = array3[cpix[1]-50:cpix[1]+50, cpix[0]-50:cpix[0]+50]

只是有一些预定义的东西(比如一个面具?)所以我可以做一个

a1 = array1[predefined_2dslice] 
a2 = array2[predefined_2dslice] 
a3 = array3[predefined_2dslice] 

这是numpy支持的吗?


Tags: a2定义a1方式切片数组解决方案中心