Я ищу векторизованный способ создания 2D-массива в каждой точке 2D-сетки и сохранения результата в 4-мерном массиве (или, по крайней мере, лучше, чем во вложенных циклах).
Более конкретно, у меня есть функция, которая создает матрицу (2D-массив) с двумя числами H(x,y). Я хотел бы использовать эту функцию для создания матрицы в каждой точке сетки (x,y) и сохранения результата в массиве 4D. Предпочтительно, если у нас есть что-то вроде
x = np.linspace(-1,1,100)
y = np.linspace(-1,1,100)
xx, yy = np.meshgrid(x,y)
result = magic_function(xx, yy, H)
, тогда форма result будет (100,100,4,4) (если H создает матрицы 4x4).