Постройте контуры из дискретных данных в matplotlib ⇐ Python

[Anonymous]

Как создать контурный график, где области должны быть дискретными (целочисленный массив вместо числа с плавающей запятой)?
Значения должны дискретно сопоставляться с индексами цвета. Вместо этого matplotlib просто масштабирует результат по всему набору цветов.
Пример:

Код: Выделить всё

import numpy as np
from matplotlib import pyplot as plt

axes = (np.linspace(-2, 2, 100), np.linspace(-2, 2, 100))
xx, yy = np.meshgrid(*axes, indexing="xy")
fig, ax = plt.subplots()
z = np.abs(xx * yy).astype(int)  # values 0, 1, 2, 3, 4
z[z==0] = 4
ax.contourf(xx, yy, z, cmap="Set1")

< img alt="дискретная цветовая шкала" src="https://i.sstatic.net/pz6l2YVf.png" />

Подробнее здесь: https://stackoverflow.com/questions/792 ... matplotlib