matplotlib: как построить 2д график-набор точек с разными цветами?

Есть массив трехмерных точек. Надо построить двухмерную карту высот, где третья координата является цветом точки. Бился, бился, а придумать ничего не получается