乘以2个numpy数组

Question

我有两个numpy数组，第一个数组A的形状是(36,)，第二个数组B是(36, 4)。想法是通过将B的4个元素中的每个元素与A的对应元素相乘的方式来对诸如A[0] * B[0]之类的对应元素进行乘积。

Answer 1

您需要向A添加新轴以启用广播。

import numpy as np

A = np.random.randint(0, 10, size=(36,4))
B = np.random.randint(0, 10, size=(36,))

A * B.reshape(-1, 1)