matplotlib - Plot 2D данные карты с использованием Python и numpy

Question

Я пытаюсь построить данные карты в формате CSV для выполнения локализации робота. У меня есть numpy 2D-массив, называемый map, который индексирует значение вероятности, которое робот находится в этой позиции сетки.

E.g. map[500, 375] = 0.75 означает, что 75% вероятность того, что робот окажется в сетке 500, 375.

В настоящее время я использую функцию matplotlib surface_plot для построения карты 3D-высоты, но я хочу использовать matplotlib для построения 2D-карты цветов, где вместо высоты вероятность представлена ​​определенным цветом.

Любые идеи о том, как я могу это сделать?

Answer 1

На самом деле я не уверен, что это то, что вам нужно, но я думаю, вы можете взглянуть на plt.imshow(). Он имеет большое количество цветовых палитр.

Например, чтобы построить что-то с помощью hot-cmap:

plt.imshow(map, origin='lower', cmap="hot", interpolation='none') 

Параметры, на самом деле не очень важно то, что origin есть (с моей точки зрения) в заблуждение, что 0,0 вверху слева по умолчанию, и вы должны указать 'lower' если вы хотите, чтобы 0,0 был внизу слева. Интерполяция может быть опущена, если вы хотите какую-то интерполяцию, но если вы не хотите ее, то вам нужно определить ее как 'none'.

Список всех возможных цветовых схем можно найти в одном из matplotlibs example codes.