python - 获取包含基本元素的数组的反向切片

Question

我正在使用 numpy 数组。我正在尝试计算一个数组，其中包含从某个中心散布的元素的乘积。我的代码目前看起来像这样，其中 psi 是一个 numpy.array():

up=psi[position:position+width]
pre_down=psi[position-width+1:position+1]
down=pre_down[::-1]
ac=up*down

是否有更优雅的 pythonic 方式使数组“向下”？ 类似

down=psi[position:position-width:-1]

当 position-width 为 0 时不起作用。

Answer 1

当您需要切片到元素 0 时，您可以使用 None 作为切片的末尾。

end = position - width - 1 if position > width else None
down = psi[position:end:-1]