在任意范围内切片n维numpy数组

时间:2020-08-13 12:45:00

标签: python arrays python-3.x numpy indexing

假设我有一个N维的numpy数组以及两个起点和终点列表。

start_points = [0, 3, 1]
end_points = [5, 8, 4]

我想做的是这样

some_slice = my_array[0:5, 3:8, 1:4]

我知道在对其进行硬编码时如何执行此操作,问题是如何在不对维度进行硬编码的情况下执行该操作

1 个答案:

答案 0 :(得分:0)

您可以这样做:

some_slice = my_array[tuple(map(slice, start_points, end_points))]