matplotlib 如何在曲线下填充彩虹色

brc7rcf0  于 2023-05-18  发布在  其他
关注(0)|答案(3)|浏览(261)

我想在曲线下填充彩虹色。实际上matplotlib.pyplot.fill_between函数可以用单一颜色填充曲线下的区域。

import matplotlib.pyplot as plt
import numpy as np
x = np.linspace(0, 100, 50) 
y = -(x-50)**2 + 2500
plt.plot(x,y)
plt.fill_between(x,y, color='green')
plt.show()

有没有一个旋钮我可以把颜色调成彩虹?谢谢。

u4dcyp6a

u4dcyp6a1#

如果你想用一系列矩形“填充”,这是很容易破解的:

import numpy as np
import pylab as plt

def rect(x,y,w,h,c):
    ax = plt.gca()
    polygon = plt.Rectangle((x,y),w,h,color=c)
    ax.add_patch(polygon)

def rainbow_fill(X,Y, cmap=plt.get_cmap("jet")):
    plt.plot(X,Y,lw=0)  # Plot so the axes scale correctly

    dx = X[1]-X[0]
    N  = float(X.size)

    for n, (x,y) in enumerate(zip(X,Y)):
        color = cmap(n/N)
        rect(x,0,dx,y,color)

# Test data    
X = np.linspace(0,10,100)
Y = .25*X**2 - X
rainbow_fill(X,Y)
plt.show()

您可以通过缩小矩形(即,使用更多点)。此外,您可以使用梯形(甚至插值多项式)来细化“矩形”。

hm2xizp9

hm2xizp92#

如果你的意思是给“color=”一些聪明的论据,恐怕就我所知,这并不存在。您可以通过为每种颜色设置一条二次线并改变偏移量来手动执行此操作。在它们之间填充正确的颜色会给予rainbowish这是一个有趣的项目来学习一些Python,但如果你不想尝试这里是一个例子:

import matplotlib.pyplot as plt
import numpy as np

x = np.linspace(0, 100, 50) 

y_old = -(x-50)**2 + 2500
for delta, color in zip([2250, 2000, 1750, 1500, 1250, 1000], ["r", "orange", "g", "b", "indigo", "violet"] ):
    y_new = -(x-50)**2 + delta
    plt.plot(x, y, "-k")
    plt.fill_between(x, y_old, y_new, color=color)
    y_old = y_new

plt.ylim(0, 2500)
plt.show()

你会发现,这看起来不像彩虹。这是因为我们使用的函数是二次函数,实际上彩虹是由不同半径的圆组成的(这里还有一个有趣的数学项目!)。这也是可绘制的matplotlib,我会尝试这个,使它,这样你可以绘制超过7种颜色在彩虹eidg图1000颜色跨越整个光谱,使它看起来真的像一个彩虹!

iszxjhcz

iszxjhcz3#

这里是一个修改后的解决方案,使用梯形而不是矩形。

import numpy as np
import pylab as plt

# a solution that uses rectangles
def rect(x,y,w,h,c):
    ax = plt.gca()
    polygon = plt.Rectangle((x,y),w,h,color=c)
    ax.add_patch(polygon)

# a solution that uses trapezoids
def polygon(x1,y1,x2,y2,c):
    ax = plt.gca()
    polygon = plt.Polygon( [ (x1,y1), (x2,y2), (x2,0), (x1,0) ], color=c )
    ax.add_patch(polygon)

def rainbow_fill(X,Y, cmap=plt.get_cmap("jet")):
    plt.plot(X,Y,lw=0)  # Plot so the axes scale correctly

    dx = X[1]-X[0]
    N  = float(X.size)

    for n, (x,y) in enumerate(zip(X,Y)):
        color = cmap(n/N)
        # uncomment to use rectangles
        # rect(x,0,dx,y,color)
        # uncomment to use trapezoids
        if n+1 == N: continue
        polygon(x,y,X[n+1],Y[n+1],color)

# Test data    
X = np.linspace(0,10,100)
Y = .25*X**2 - X
rainbow_fill(X,Y)
plt.show()

相关问题