numpy：数组数组乘以整数数组等于数组

Question

z有没有一种不使用循环来计算数组的 Pythonic 方法？import numpy as npx = np.array([[1, 2, 3], [6, 7, 8]])y = np.array([5, 8])z = np.array([x[i] * y[i] for i in range(0, len(x))])

Answer 1

你可以做：

x * np.expand_dims(y, 1)

expand_dims当乘以 (shape 2×3) 时，乘法将根据需要广播 ( xshape 2×1 )的结果。

结果：

array([[ 5, 10, 15],
       [48, 56, 64]])

Answer 2

您将需要适当的暗淡y(shape ) 与(shape )2,1相乘，这里有一些其他方法可以将单个额外的暗淡添加到x2,3y

x * y[:, None]

array([[ 5, 10, 15],

       [48, 56, 64]])

x * y[:, np.newaxis]

array([[ 5, 10, 15],

       [48, 56, 64]])