numpy：将每行除以一个向量元素

Question

假设我有一个numpy数组：

data = np.array([[1,1,1],[2,2,2],[3,3,3]])

我有一个对应的“向量”：

vector = np.array([1,2,3])

我如何data沿每一行进行减法或除法运算，所以结果是：

sub_result = [[0,0,0], [0,0,0], [0,0,0]]
div_result = [[1,1,1], [1,1,1], [1,1,1]]

长话短说：如何使用对应于每一行的1D标量数组在2D数组的每一行上执行操作？

Answer 1

干得好。您只需要与广播结合使用None（或np.newaxis）：

In [6]: data - vector[:,None]
Out[6]:
array([[0, 0, 0],
       [0, 0, 0],
       [0, 0, 0]])

In [7]: data / vector[:,None]
Out[7]:
array([[1, 1, 1],
       [1, 1, 1],
       [1, 1, 1]])