Умножить numpy ndarray на массив 1d вдоль заданной оси ⇐ Python

[Anonymous]

Кажется, я заблудился в чем-то потенциально глупом.
У меня есть n-мерный массив numpy, и я хочу умножить его на вектор (1d-массив) по некоторому измерению (которое может измениться!).
В качестве примера, скажем, я хочу умножить 2D-массив на 1D-массив вдоль оси 0 первого массива, я могу сделать что-то вроде этого:

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

a=np.arange(20).reshape((5,4))
b=np.ones(5)
c=a*b[:,np.newaxis]

Просто, но я хотел бы распространить эту идею на n-мерности (для a, а b всегда равно 1d) и на любую ось. Другими словами, я хотел бы знать, как создать срез с np.newaxis в нужном месте. Предположим, что a равно 3d, и я хочу умножить вдоль оси = 1, я хотел бы создать срез, который бы правильно давал:

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

c=a*b[np.newaxis,:,np.newaxis]

Т.е. учитывая количество измерений (скажем, 3) и ось, вдоль которой я хочу умножить (скажем, ось = 1), как мне сгенерировать и передать срез:

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

np.newaxis,:,np.newaxis

Спасибо.

Подробнее здесь: https://stackoverflow.com/questions/300 ... given-axis