Построение двумерного массива с помощью mplot3d

У меня есть массив 2D numpy, и я хочу построить его в 3D. Я слышал о mplot3d, но не могу нормально работать

Вот пример того, что я хочу сделать. У меня есть массив с размерами (256,1024 ). Он должен построить трехмерный график, где ось x от 0 до 256, ось y от 0 до 1024, а ось z графика отображает значение массива для каждой записи.

Как мне это сделать?