在matplotlib中,如何填充由两组不同数组定义的两条曲线?

2024-04-27 15:49:47 发布

您现在位置:Python中文网/ 问答频道 /正文

我有两条由两组数组定义的曲线:(x1, y1)(x2, y2),我想用多边形填充它们之间。所有数组的长度相同,但x1x2包含不同的值。在

plt.fill_between(x, y1, y2)要求两条曲线共享同一个x数组。在

我该怎么做fill_between(x1, y1, x2, y2)?在

例如,如果:

x1 = np.array([1.0, 2.0, 3.0, 4.0, 5.0])y1 = np.array([3.0, 2.0, 3.0, 2.0, 3.0])定义第一条曲线

以及

x2 = np.array([1.5, 2.5 ,3.5 ,4.5 , 5.5])和{}定义第二个。在

如何使用四个多边形填充曲线(x1, y1)(x2, y2)之间的颜色(左右边界不必垂直)?在

为了澄清,四个多边形(A、B、C、D)将具有坐标:

A: [(1.0, 3.0), (1.5, 5.0), (2.5, 6.0), (2.0, 2.0)]
B: [(2.0, 2.0), (2.5, 6.0), (3.5, 7.0), (3.0, 3.0)]
C: [(3.0, 3.0), (3.5, 7.0), (4.5, 8.0), (4.0, 2.0)]
D: [(4.0, 2.0), (4.5, 8.0), (5.5, 9.0), (5.0, 3.0)]

Tags: 定义颜色npplt数组between多边形array
2条回答

您可以使用多边形面片绘制填充两条曲线之间空间的四边形-唯一棘手的是生成定义多边形的5个点,但是(ab)使用zip它可以完成。。。此外,您还需要知道如何在绘图上放置多边形,但是当您知道matplotlib.collections.PatchCollection和{}时就很容易了

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

x1 = np.linspace(0,6,21) ; y1 = np.sin(x1)
x2 = x1+0.28 ; y2 = np.cos(x2)

fig, ax = plt.subplots()                                                         
ax.plot(x1, y1, x2, y2)
patches = [Polygon(poly) for poly in (
              [p0,p1,p2,p3,p0] for p0,p1,p2,p3 in 
                    zip(zip(x1,y1),zip(x1[1:],y1[1:]),zip(x2[1:],y2[1:]),zip(x2,y2)))
ax.add_collection(PatchCollection(patches, alpha=0.6))

a sample plot

正如你所见,它不是完美的,但也许它足够好。。。在

我很难明确地解释你的问题,但我认为你只想用fill,类似于:

import numpy as np
import matplotlib.pyplot as plt

x1 = np.array([1.0, 2.0, 3.0, 4.0, 5.0])
y1 = np.array([3.0, 2.0, 3.0, 2.0, 3.0])

x2 = np.array([1.5, 2.5, 3.5, 4.5, 5.5])
y2 = np.array([5.0, 6.0, 7.0, 8.0, 9.0])

plt.plot(x1, y1, 'o')
plt.plot(x2, y2, 'x')

plt.fill(
    np.append(x1, x2[::-1]),
    np.append(y1, y2[::-1]),
)

会给你

this plot

相关问题 更多 >