热门文章
 当前位置：技术文章 > 编程面试题 >

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

问题内容：

我试图找到一个整齐的解决方案，但我正在以相同的方式切片相同形状的几个2D数组。我通过定义一个包含’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支持的吗？


问题答案：

是的，您可以使用numpy.s_：

例：

>>> a = np.arange(10).reshape(2, 5)
>>> 
>>> m = np.s_[0:2, 3:4]
>>> 
>>> a[m]
array([[3],
       [8]])

在这种情况下：

my_slice = np.s_[cpix[1]-50:cpix[1]+50, cpix[0]-50:cpix[0]+50]

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

您还可以使用numpy.r_以便将切片对象平移到沿第一个轴的串联。