python - 将 2 个 numpy 数组相乘

Question

我有两个 numpy 数组，第一个数组的形状A是(36,)，第二个B是(36, 4). 这个想法是将对应的元素相乘，就像A[0] * B[0]B 的 4 个元素中的每一个都与 A 的对应元素相乘一样。

标签： pythonnumpy

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)