Numpy:沿移动轴执行原位操作

一种可能性是移动轴,将您想要的轴(根据您的标准)放在新视图中的第一位 a2 然后 a2[i,:,:] 与相同 arr[:,i,:] (例如)如果正确的轴是1。另外,这样,您可以使用 ... 符号,以替换所有 : 所以 a2[i,...] .

# Move axis whose size is 100 in 1st position (index 0) in a2 view
# (a2 is a view of arr. So modifying its contents change arr content).
# Note, if the 1st axis is already of size 100 this will do nothing
a2 = np.moveaxis(arr, arr.shape.index(100), 0) 

for i, (low, high) in enumerate(values):
    a2[i, ...] *= high - low
    # Note this `...` works whatever the dimension, even 1.

您可以动态创建索引。索引集是一个元组,并且 : 是 slice(None) .

在维度中查找与值长度相同的索引。然后更新循环中索引列表列表中的索引。

indexes = [slice(None)] * arr.ndims
variable_pos = arr.shape.index(len(values))
for i, (low, high) in enumerate(values):
    indexes[variable_pos] = i
    arr[tuple(indexes)] *= high - low