python - 你能在Python中增加一个切片吗？

Question

我正在通过移植我在 matlab/octave 中编写的一些代码来尝试 numpy。在 matlab 中，我可以定义相当于 python 切片，然后根据需要递增它。例如，在我的 matlab 代码中，我有

HXx_range = 1:NHXx;
HXy_range = 1:NHXy;

blah blah blah

Hx(HXx_range, HXy_range) = Da(Hx_media(HXx_range, HXy_range)).*Hx(HXx_range, HXy_range) + Db(Hx_media(HXx_range, HXy_range)).*(Ez(HXx_range,HXy_range) -   Ez(HXx_range,**HXy_range+1**));
Hy(HYx_range, HYy_range) = Da(Hy_media(HYx_range, HYy_range)).*Hy(HYx_range, HYy_range) + Db(Hy_media(HYx_range, HYy_range)).*(Ez(**HYx_range+1**,HYy_range) - Ez(HYx_range,HYy_range));
Ez(EZx_range, EZy_range) = Ca(Ez_media(EZx_range, EZy_range)).*Ez(EZx_range, EZy_range) + Cb(Ez_media(EZx_range, EZy_range)).*(Hy(EZx_range,EZy_range) - Hy(**EZx_range-1**,EZy_range) + Hx(EZx_range,**EZy_range-1**) - Hx(EZx_range,EZy_range));

“**”中的术语(如“HXy_range+1”)是关键部分； HXy_range+1 等于 2:(NHXy+1)。在Python中，我可以用类似的方式定义一个切片:

HXx_range = slice(0, NHXx)

但是，HXx_range+1 给了我一个错误。当然，我可以为此制作一个新切片，但它不是那么干净。有办法解决这个问题吗？

谢谢。

Answer 1

如果您将 HXy_range 定义为 numpy 数组，那么您可以根据需要递增它。如果您希望将其用作切片，则可以形成 slice(*HXy_range):

In [26]: HXy_range = np.array([1,10])

In [27]: HXy_range+1
Out[27]: array([ 2, 11])

In [28]: slice(*(HXy_range+1))
Out[30]: slice(2, 11, None)