python - 划分矩阵的向量列

Question

假设矩阵

X = np.array([[ 1,  2,  3],
              [ 4,  5,  6],
              [ 7,  8,  9],
              [10, 11, 12]])

和向量 divisor = [5, 10, 15] .有没有使用 numpy 的方法划分 X 的第一列来自 divisor[0] ， X的第二列来自 divisor[1]和 X的第二列来自 divisor[3] ?

Answer 1

只需将它们分开，让 numpy 处理广播:

X/divisor

输出:

[[0.2 0.2 0.2]
 [0.8 0.5 0.4]
 [1.4 0.8 0.6]
 [2.  1.1 0.8]]

如果要划分行(而不是列)，请使用:

X/divisor[:,None]

它为除数添加了一个维度，因此 numpy 可以正确地广播它。