установка цветового диапазона в коллекции патчей matplotlib

Я рисую PatchCollection в matplotlib с координатами и значениями цвета патча, считанными в из файла.

Проблема в том, что matplotlib автоматически масштабирует цветовой диапазон до минимального/максимального значений данных. Как я могу вручную установить цветовой диапазон? Например. если мой диапазон данных составляет 10-30, но я хочу масштабировать его до цветового диапазона 5-50 (например, для сравнения с другим графиком), как я могу это сделать?

Мои команды построения графиков выглядят почти так же, как в примере кода API: patch_collection.py

colors = 100 * pylab.rand(len(patches))
p = PatchCollection(patches, cmap=matplotlib.cm.jet, alpha=0.4)
p.set_array(pylab.array(colors))
ax.add_collection(p)
pylab.colorbar(p)

pylab.show()

python matplotlib patch
person Mark W    schedule 17.05.2011    source источник

Ответы (1)


arrow_upward
32
arrow_downward

Используйте p.set_clim([5, 50]), чтобы установить минимумы и максимумы цветового масштабирования в случае вашего примера. Все в matplotlib, имеющее цветовую карту, имеет get_clim и set_clim методы.

В качестве полного примера:

import matplotlib
import matplotlib.pyplot as plt
from matplotlib.collections import PatchCollection
from matplotlib.patches import Circle
import numpy as np

# (modified from one of the matplotlib gallery examples)
resolution = 50 # the number of vertices
N = 100
x       = np.random.random(N)
y       = np.random.random(N)
radii   = 0.1*np.random.random(N)
patches = []
for x1, y1, r in zip(x, y, radii):
    circle = Circle((x1, y1), r)
    patches.append(circle)

fig = plt.figure()
ax = fig.add_subplot(111)

colors = 100*np.random.random(N)
p = PatchCollection(patches, cmap=matplotlib.cm.jet, alpha=0.4)
p.set_array(colors)
ax.add_collection(p)
fig.colorbar(p)

fig.show()

введите здесь описание изображения

Теперь, если мы просто добавим p.set_clim([5, 50]) (где p — коллекция патчей) где-нибудь перед вызовом fig.show(...), мы получим следующее: enter

person Joe Kington    schedule 17.05.2011