Как умножить массивы разных форм в Python?

Question

Я пытаюсь умножить два массива двух разных форм:

np.shape(a) = (20,37,2,121,240)

np.shape(b) = (121,240)

Я хотел бы умножить a*b, где только значения в двух одинаковых элементах (121,240) умножаются вместе.Первые 3 элемента массива "a" (20,37,2) останутся прежними.

Есть ли более простой способ сделать это, кроме использования циклов for?

Answer 1

Numpy вещает автоматически, поэтому вы можете просто сделать

a * b

Пример:

In [1]: import numpy as np

In [2]: a = np.ones((20,37,2,121,240))

In [3]: b = np.ones(((121,240)))

In [4]: (a * b).shape
Out[4]: (20, 37, 2, 121, 240)