Matplotlib - Контурный график с одним значением

Я хочу построить контурный график некоторых данных, но возможно, что все значения в поле имеют одно и то же значение. Это вызывает ошибку в matplotlib, что имеет смысл, поскольку на самом деле нет контура, который нужно создавать. Например, если вы запустите приведенный ниже код, вы получите сообщение об ошибке, но удалите второе определение zi , и оно будет работать должным образом.

Как я могу построить «контурный» график для некоторых данных если это единое поле? Я хочу, чтобы он выглядел так же, как обычный контурный график (чтобы поле было заполнено каким-то цветом и отображалась цветная полоса сбоку. Цветовая полоса могла быть однородного цвета или все еще отображать диапазон из 15 цветов, я не неважно).

Код:

from numpy        import array
import matplotlib.pyplot as plt

xi = array([0., 0.5, 1.0])
yi = array([0., 0.5, 1.0])
zi = array([[0., 1.0, 2.0],
            [0., 1.0, 2.0],
            [0., 1.0, 2.0]])
zi = array([[1.0, 1.0, 1.0],
            [1.0, 1.0, 1.0],
            [1.0, 1.0, 1.0]])

CS = plt.contour(xi, yi, zi, 15, linewidths=0.5, colors='k')
CS = plt.contourf(xi, yi, zi, 15, cmap=plt.cm.jet)
plt.colorbar()
plt.show()