Сетка координат в NumPy

Функция meshgrid по двум рядам координат возвращает пару матриц: в одной повторяются значения первой оси, в другой второй. Так удобно считать функции от двух переменных на всей сетке сразу.

При длине 3 по оси x и 2 по оси y обе матрицы обычно имеют форму (2, 3). По умолчанию первая ось идёт вдоль строк, как на привычном графике. Построим сетку и выведем формы:

import numpy as np

x = np.array([0, 1, 2])
y = np.array([10, 20])
grid_x, grid_y = np.meshgrid(x, y)
print(np.shape(grid_x)) # выведет (2, 3)
print(np.shape(grid_y)) # выведет (2, 3)

По осям [1, 2] и [3, 4, 5] постройте сетку и выведите формы обеих матриц.

По осям [0, 1, 2] и [5] постройте сетку и выведите форму (1, 3) у матрицы координат первой оси.