Как вы увеличиваете цветовую гамму для контура-графика пиплотов

Question

Я пытаюсь построить функцию ошибки как контур. Используемое мной приближение имеет две переменные, каждая из которых выдает среднюю ошибку:

Я сейчас пытаюсь построить это как контур:

def contour(pa, pm, data):

    print("Plotting Contour")
    plt.xlabel("alpha")
    plt.ylabel("m")

    plt.contourf(pa, pm, data)
    plt.colorbar()

    plt.show()

В результате получается контурная диаграмма:

К сожалению, я теряю много деталей в долине. Как увеличить цветовую гамму или есть возможность просто закрасить все, что выше определенного порога, желтым?

Answer 1

Используйте plt.pcolor , vmin и vmax . Прочитайте об этом в документации - объясняет это проще, чем я могу.