градиент цвета matplotlib в патчах?

Я хотел бы создать эллипсы в matplotlib с цветом заливки, который имеет значение альфа (непрозрачность), которое зависит от радиуса;

например, 2D Гауссова.

есть ли способ сделать это?

можно достаточно легко создавать прямоугольные участки с цветовыми градиентами (например,градиент facecolor matplotlib гистограмма и этой) но я не могу понять, как сделать то же самое для кругов/эллипсов.

2 ответов


вот пример функции, использующей идею из поста Алекса

import matplotlib.pyplot as plt,numpy as np

def gauplot(centers, radiuses, xr=None, yr=None):
        nx, ny = 1000.,1000.
        xgrid, ygrid = np.mgrid[xr[0]:xr[1]:(xr[1]-xr[0])/nx,yr[0]:yr[1]:(yr[1]-yr[0])/ny]
        im = xgrid*0 + np.nan
        xs = np.array([np.nan])
        ys = np.array([np.nan])
        fis = np.concatenate((np.linspace(-np.pi,np.pi,100), [np.nan]) )
        cmap = plt.cm.gray
        cmap.set_bad('white')
        thresh = 3
        for curcen,currad in zip(centers,radiuses):
                curim=(((xgrid-curcen[0])**2+(ygrid-curcen[1])**2)**.5)/currad*thresh
                im[curim<thresh]=np.exp(-.5*curim**2)[curim<thresh]
                xs = np.append(xs, curcen[0] + currad * np.cos(fis))
                ys = np.append(ys, curcen[1] + currad * np.sin(fis))
        plt.imshow(im.T, cmap=cmap, extent=xr+yr)
        plt.plot(xs, ys, 'r-')

и вот что вы получаете, когда вы запустите

    gauplot([(0,0), (2,3), (5,1), (6, 7), (6.1, 6.1)], [.3,. 4, .5, 1, .4], [-1,10], [-1,10])
             #           centers of circles           # radii of circles#

plot


не думаю matplotlib в настоящее время поддерживает градиентные заливки для патчи - см. этот адрес электронной почты.

john> Здравствуйте, я пытаюсь установить панель (исправленную серию прямоугольников) с рисунком заливки вместо сплошного цвета. Есть ли простой способ сделать это в matplotlib?
Джон> я думаю о чем-то вроде QBrush Qt, который имеет крест, вертикаль, плотность и т. д. узоры.

В настоящее время нет поддержки для этого-это не было бы слишком трудно добавьте для бэкэндов, которые поддерживают такие вещи. В принципе, нам нужно чтобы указать API для него и добавить поддержку бэкэндов. Я был хотите добавить градиентные заливки для патчей (например, полигоны, прямоугольники) и хорошо бы сделать оба сразу.


вместо использования патчей вы можете создать сетку, рассчитать цвета с помощью функции, а затем использовать imshow с интерполяцией:

# Taken from http://matplotlib.sourceforge.net/examples/pylab_examples/layer_images.html

def func3(x,y):
    return (1- x/2 + x**5 + y**3)*exp(-x**2-y**2)

# make these smaller to increase the resolution
dx, dy = 0.05, 0.05

x = arange(-3.0, 3.0, dx)
y = arange(-3.0, 3.0, dy)
X,Y = meshgrid(x, y)

xmin, xmax, ymin, ymax = amin(x), amax(x), amin(y), amax(y)
extent = xmin, xmax, ymin, ymax

fig = plt.figure(frameon=False)

Z2 = func3(X, Y)

im2 = imshow(Z2, cmap=cm.jet, alpha=.9, interpolation='bilinear', extent=extent)

show()

это приведет к следующему (игнорировать клетчатый фон):

http://matplotlib.sourceforge.net/_images/layer_images.png