索引numpy数组的中间?

2024-05-21 02:15:14 发布

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

要索引numpy数组的中间点,可以执行以下操作:

x = np.arange(10)
middle = x[len(x)/4:len(x)*3/4]

有没有索引数组中间的速记法?e、 例如,n2n元素接近len(x)/2?有一个很好的n维版本吗?


Tags: 版本numpy元素middlelennp数组arange
1条回答
网友
1楼 · 发布于 2024-05-21 02:15:14

正如cge所说,最简单的方法是将其转换为lambda函数,如下所示:

x = np.arange(10)
middle = lambda x: x[len(x)/4:len(x)*3/4]

或者n维的方式是:

middle = lambda x: x[[slice(np.floor(d/4.),np.ceil(3*d/4.)) for d in x.shape]]

相关问题 更多 >