Colorbar не соответствует контурному графику ⇐ Python

[Anonymous]

Я хочу сделать контурный сюжет только с 2 цвета: синий, если значение ниже порога, красный, если он выше. Я также добавил цвета для большей ясности. Вот кодовая часть, я добровольно упростил ее для этого поста: < /p>
from matplotlib import colors, cm
import numpy as np
import matplotlib.pyplot as plt
import matplotlib.tri as tri

np.random.seed(19680801)
npts = 200
ngridx = 100
ngridy = 200
x = np.random.uniform(-2, 2, npts)
y = np.random.uniform(-2, 2, npts)
z = x * np.exp(-x**2 - y**2)

xi = np.linspace(-2.1, 2.1, ngridx)
yi = np.linspace(-2.1, 2.1, ngridy)

triang = tri.Triangulation(x, y)
interpolator = tri.LinearTriInterpolator(triang, z)

Xi, Yi = np.meshgrid(xi, yi)
zi = interpolator(Xi, Yi)

fig = plt.figure()
ax = fig.subplots()

colormapList = ['blue', 'red']
colorThreshold = np.mean(z)

norm = colors.BoundaryNorm([colorThreshold-1e-3, colorThreshold], ncolors = 2, clip = True)
cmap = colors.LinearSegmentedColormap.from_list('name', colormapList, N = 2)

ax.contour(xi, yi, zi, norm = norm, cmap = cmap)

mappable = cm.ScalarMappable(norm = norm, cmap = cmap)

cbar = fig.colorbar(mappable, ax = ax)
cbar.set_ticks([np.min(zi), colorThreshold, np.max(zi)])
cbar.set_ticklabels([np.min(zi), colorThreshold, np.max(zi)])
< /code>
Проблема, которая у меня есть, состоит в том, что цветовая бара должна отображаться синий ниже порога и красный над порогом, но он показывает белый и синий под порогом, затем белый над порогом. По какой -то причине это также показывает 4 -й неназванный тик, когда я прошу только 3 тика. Вот пример:

Что я упускаю?

Подробнее здесь: https://stackoverflow.com/questions/797 ... tour-graph