动态索引具有未知维数和范围的 ndarray

Question

我有未知形状的data数组和用于切片data的数组bounds 。 此代码适用于 3D data ，但有没有办法将其推广到 N-dim？

for b in bounds:
    l0, u0 = b[0]
    l1, u1 = b[1]
    l2, u2 = b[2]
    a = data[l0:u0, l1:u1, l2:u2]
    print(a)

尝试使用范围 python object 作为索引，但没有用。

数据示例：

data2D = np.arange(2*3).reshape((2, 3))
data3D = np.arange(2*3*4).reshape((2, 3, 4))

对应边界：

bounds2D = np.array([[[0, 2], [0, 2]], [[0, 2], [1, 3]]])
bounds3D = np.array(
        [
            [[0, 2], [0, 2], [0, 2]],
            [[0, 2], [0, 2], [2, 4]],
            [[0, 2], [1, 3], [0, 2]],
            [[0, 2], [1, 3], [2, 4]],
        ],
    )

Answer 1

您可以使用切片function 从bounds中的每个元素创建一个切片。 然后将这些切片收集到一个元组中，并使用它正确地恢复数组中想要的项目。 您可以按如下方式调整代码：

import numpy as np


# The dimension of the slices is equal to the 
# one specified by the bounds provided
def create_slices(bounds):
    slices = list()

    # Take a single item of the bounds and create corresponding slices
    for b in bounds:
        # Slices are collected inside a single tuple
        slices.append(tuple([slice(l, u) for l, u in b]))
    return slices
   
# 4D example data   
data4D = np.arange(2*3*4*5).reshape((2,3,4,5))

# Bounds array for 4D data
bounds4D = np.array(
        [
            [[0, 2], [0, 2], [0, 2], [0, 2]],
            [[0, 2], [0, 2], [0, 2], [2, 4]],
            [[0, 2], [1, 3], [2, 4], [0, 2]],
            [[0, 2], [1, 3], [2, 4], [2, 4]],
        ],
   )


slices = create_slices(bounds4D)

# Each element of slices is a single slice that can be used on 
# the corresponding data array
for single_slice in slices:
    a = data4D[single_slice]
    print("Slice", a)