Совместное растягивание в NumPy

Растянуть два массива к одной форме позволяет функция broadcast_arrays. Она принимает несколько массивов и возвращает их же, но уже согласованные по правилам broadcasting: каждый участник повторяется там, где его ось короче.

Столбец из трёх чисел и один ряд из двух можно привести к общей таблице (3, 2). Получим пару массивов и выведем их формы:

import numpy as np

col = np.array([[1], [2], [3]])
row = np.array([10, 20])
wide_col, wide_row = np.broadcast_arrays(col, row)
print(np.shape(wide_col)) # выведет (3, 2)
print(np.shape(wide_row)) # выведет (3, 2)

Согласуйте столбец [[0], [1]] и ряд [10, 20, 30] и выведите форму обоих результатов.

Согласуйте столбец [[1], [2], [3]] и ряд [[10, 20, 30]] и выведите форму (3, 3) у каждого из двух массивов.