Произвольный нарезка трафарета в Numpy ⇐ Python

[Anonymous]

Есть ли простой синтаксис для создания ссылок на произвольное количество соседних элементов массива в Numpy? Ширина трафарета, например, < /p>

Код: Выделить всё

import numpy as np

x = np.arange(8)

# Hard-coded stencil width of 3
x_neighbours = (
x[  :-2]
x[ 1:-1]
x[ 2:  ]
)

< /code>
Однако моя попытка произвольной ширины трафареты не особенно читаемо < /p>

nStencil = 3
x_neighbours = (
x[indexStart:indexStop]
for indexStart, indexStop in zip(
(None, *range(1,nStencil)),
(*range(1-nStencil,0), None),
)
)

Есть лучший подход?

Подробнее здесь: https://stackoverflow.com/questions/796 ... g-in-numpy