python - python中二维和一维数组的元素明智乘法

Question

假设我有两个 numpy 数组:

import numpy as np
x = np.array([[1,2,3], [4,5,6], [7,8,9]])
y = np.array([-1, 1, -1])

我想以得到 z 的方式将 x 和 y 相乘:

z = np.array([[-1,2,-3], [-4,5,-6], [-7,8,-9]])

换句话说，如果 y 的元素 j 为 -1，则 x 的第 j 行的所有元素都乘以 -1。如果 y 的元素 k 为 1，则 x 的第 j 行的所有元素都乘以 1。

我该怎么做？

Answer 1

只需使用乘法运算符:

x * y
Out[6]: 
array([[-1,  2, -3],
       [-4,  5, -6],
       [-7,  8, -9]])