Умножение 2 numpy массивов

Question

У меня есть два массива numpy, форма первого массива A равна (36,), а второго B равна (36, 4). Идея состоит в том, чтобы умножить соответствующие элементы, такие как A[0] * B[0], таким образом, чтобы каждый из 4 элементов B умножался на соответствующий элемент A.

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)