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

Question

z有没有一种不使用循环来计算数组的 Pythonic 方法？

import numpy as np

x = 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,3yx * y[:, None]array([[ 5, 10, 15],       [48, 56, 64]])x * y[:, np.newaxis]array([[ 5, 10, 15],       [48, 56, 64]])