如何将列X的所有条目乘以常量，其中X是变量

Question

给定X和具有D维度的numpy数组，如何将列X的所有条目乘以C.例如，考虑下面的3D情况：

如果X = 0，则所需命令为

M[1,::]*=C

如果X = 1，则所需命令为

M[:,1,:]*=C

如果X = 2，则所需命令为

M[::,1]*=C

我如何对其进行编码以使其一般有效？ 谢谢！

Answer 1

一个简单的方法是

M.swapaxes(0, X)[1] *= C