python - 如何将 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